diff --git a/python/sglang/test/runners.py b/python/sglang/test/runners.py index 61781fea21de..fa737338d7f1 100644 --- a/python/sglang/test/runners.py +++ b/python/sglang/test/runners.py @@ -234,7 +234,7 @@ def _forward_gme_qwen2_vl( **kwargs, ) -> torch.Tensor: if inputs_embeds is None: - inputs_embeds = self.model.model.embed_tokens(input_ids) + inputs_embeds = self.model.model.get_input_embeddings()(input_ids) if pixel_values is not None: pixel_values = pixel_values.type(self.model.visual.get_dtype()) image_embeds = self.model.visual(