From 5d712571afd87f5cc1b6c6c25feaac5c706f3712 Mon Sep 17 00:00:00 2001 From: Jeff Cook Date: Thu, 12 Dec 2024 11:09:20 -0700 Subject: [PATCH] [Bugfix] Quick fix to make Pixtral-HF load correctly again after 39e227c7ae. (#11024) --- vllm/model_executor/models/llava.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/vllm/model_executor/models/llava.py b/vllm/model_executor/models/llava.py index 65c6bd07bfff0..53eef72dd5f91 100644 --- 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):