revert

vllm-project · Dec 11, 2024 · c5c5aa8 · c5c5aa8
1 parent e1fb91e
commit c5c5aa8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/vllm/model_executor/models/llava.py b/vllm/model_executor/models/llava.py
@@ -218,11 +218,12 @@ def _get_dummy_mm_kwargs(
         image_processor = hf_processor.image_processor  # type: ignore
         hf_inputs = image_processor.preprocess(data['image'],
                                                return_tensors="pt")
-        if 'is_pixtral' not in hf_inputs:
-            hf_inputs['is_pixtral'] = isinstance(hf_processor,
-                                                 PixtralProcessor)
+        is_pixtral = isinstance(hf_processor, PixtralProcessor)
 
-        return MultiModalKwargs(**hf_inputs)
+        return MultiModalKwargs(
+            **hf_inputs,
+            is_pixtral=torch.tensor(is_pixtral),
+        )
 
 
 class LlavaLikeConfig(Protocol):