[Bugfix] Quick fix to make Pixtral-HF load correctly again after 39e227c7. (#11024)

5d712571 · Jeff Cook · GitHub · d4d5291c · 5d712571
Unverified Commit 5d712571 authored Dec 12, 2024 by Jeff Cook Committed by GitHub Dec 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 5 deletions

vllm/model_executor/models/llava.py vllm/model_executor/models/llava.py +1 -5

No files found.
--- a/vllm/model_executor/models/llava.py
+++ b/vllm/model_executor/models/llava.py
@@ -218,12 +218,8 @@ class LlavaProcessor(BaseMultiModalProcessor):
        image_processor = hf_processor.image_processor  # type: ignore
        hf_inputs = image_processor.preprocess(data['image'],
                                               return_tensors="pt")
-        is_pixtral = isinstance(hf_processor, PixtralProcessor)

-        return MultiModalKwargs(
-            **hf_inputs,
-            is_pixtral=torch.tensor(is_pixtral),
-        )
+        return MultiModalKwargs(**hf_inputs)


 class LlavaLikeConfig(Protocol):