xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 21:15:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] fix imgs_pos in hunyuan_vl (#29879)

Browse Source 
Co-authored-by: Isotr0py <mozf@mail2.sysu.edu.cn>
This commit is contained in:
JackieWu 2025-12-03 14:20:37 +08:00 committed by GitHub
parent c719c40540
commit 0bec63fa31
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/transformers_utils/processors/hunyuan_vl.py
View File

@ -123,7 +123,7 @@ class HunYuanVLProcessor(ProcessorMixin):
attention_mask = input_ids.ne(self.pad_id) attention_mask = input_ids.ne(self.pad_id)
text_inputs["attention_mask"] = attention_mask text_inputs["attention_mask"] = attention_mask
text_inputs["imgs_pos"] = [self.get_imgs_pos(input_ids)] text_inputs["imgs_pos"] = [self.get_imgs_pos(e) for e in input_ids]
# image_inputs["imgs"] = [[image_inputs["pixel_values"]]] # image_inputs["imgs"] = [[image_inputs["pixel_values"]]]
return_tensors = kwargs.pop("return_tensors", None) return_tensors = kwargs.pop("return_tensors", None)