Use get_input_embeddings

Author: rahul-tuli

vllm/model_executor/models/llama_eagle3.py

@@ -249,8 +249,7 @@ def get_input_embeddings(
         multimodal_embeddings: Optional[NestedTensors] = None,
         is_multimodal: Optional[torch.Tensor] = None,
     ) -> torch.Tensor:
-        # The llama3 drafter only processes text embeddings
-        return self.model.embed_tokens(input_ids)
+        return self.model.get_input_embeddings(input_ids)
 
     def forward(
         self,