From e4806d973acba6550dd061830471b19e8c54e692 Mon Sep 17 00:00:00 2001 From: duke <108673086+iwzbi@users.noreply.github.com> Date: Mon, 15 Dec 2025 18:38:29 +0800 Subject: [PATCH] [BugFix] Add embed_input_ids method to make QWenLMHeadModel a vllm model (#30674) Signed-off-by: root Co-authored-by: root --- vllm/model_executor/models/qwen.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/vllm/model_executor/models/qwen.py b/vllm/model_executor/models/qwen.py index 492ba2fb12145..61a6e67805d6a 100644 --- a/vllm/model_executor/models/qwen.py +++ b/vllm/model_executor/models/qwen.py @@ -281,6 +281,9 @@ class QWenBaseModel(nn.Module): self.transformer.make_empty_intermediate_tensors ) + def embed_input_ids(self, input_ids: torch.Tensor) -> torch.Tensor: + return self.transformer.wte(input_ids) + def compute_logits( self, hidden_states: torch.Tensor,