Skip to content

GitLab

Unverified Commit 5d712571 authored by Jeff Cook's avatar Jeff Cook Committed by GitHub
Browse files

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

parent d4d5291c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 5 deletions
vllm/model_executor/models/llava.py
View file @ 5d712571
...@@ -218,12 +218,8 @@ class LlavaProcessor(BaseMultiModalProcessor): ...@@ -218,12 +218,8 @@ class LlavaProcessor(BaseMultiModalProcessor):
image_processor = hf_processor.image_processor # type: ignore image_processor = hf_processor.image_processor # type: ignore
hf_inputs = image_processor.preprocess(data['image'], hf_inputs = image_processor.preprocess(data['image'],
return_tensors="pt") return_tensors="pt")
is_pixtral = isinstance(hf_processor, PixtralProcessor)
return MultiModalKwargs( return MultiModalKwargs(**hf_inputs)
**hf_inputs,
is_pixtral=torch.tensor(is_pixtral),
)
class LlavaLikeConfig(Protocol): class LlavaLikeConfig(Protocol):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment