xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-30 19:31:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jeff Cook 2024-12-12 11:09:20 -07:00 committed by GitHub
parent d4d5291cc2
commit 5d712571af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/model_executor/models/llava.py
View File

@ -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):