diff --git a/vllm_ascend/patch/worker/patch_multimodal_merge.py b/vllm_ascend/patch/worker/patch_multimodal_merge.py index 35be3cf3..6d62a317 100644 --- a/vllm_ascend/patch/worker/patch_multimodal_merge.py +++ b/vllm_ascend/patch/worker/patch_multimodal_merge.py @@ -24,8 +24,8 @@ from vllm.multimodal import NestedTensors def _merge_multimodal_embeddings( inputs_embeds: torch.Tensor, - is_multimodal: torch.Tensor, multimodal_embeddings: NestedTensors, + is_multimodal: torch.Tensor, ) -> torch.Tensor: """ Merge ``multimodal_embeddings`` into ``inputs_embeds`` by overwriting the