Skip to content

Commit

Permalink
[Bugfix] Quick fix to make Pixtral-HF load correctly again after 39e227c
Browse files Browse the repository at this point in the history
  • Loading branch information
sjuxax authored Dec 12, 2024
1 parent d4d5291 commit 5d71257
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions vllm/model_executor/models/llava.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,12 +218,8 @@ def _get_dummy_mm_kwargs(
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):
Expand Down

0 comments on commit 5d71257

Please sign in to comment.