Skip to content

Commit 2f055ec

Browse files
Isotr0pygemini-code-assist[bot]
authored andcommitted
[Bugfix] Fix incorrect channel order for idefics3 in edge case (#29881)
Signed-off-by: Isotr0py <[email protected]> Signed-off-by: Isotr0py <[email protected]> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> (cherry picked from commit 0ec8422)
1 parent 6a61085 commit 2f055ec

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/model_executor/models/idefics3.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -338,6 +338,7 @@ def _call_hf_processor(
338338
prompt_ids = self._apply_hf_processor_tokens_only(prompt_ids)
339339
return BatchFeature(dict(input_ids=[prompt_ids]), tensor_type="pt")
340340

341+
mm_kwargs = {"input_data_format": "channels_last", **mm_kwargs}
341342
processed_outputs = super()._call_hf_processor(
342343
prompt,
343344
mm_data,

0 commit comments

Comments
 (0)