Skip to content

GitLab

Unverified Commit 0bec63fa authored by JackieWu's avatar JackieWu Committed by GitHub
Browse files

[BugFix] fix imgs_pos in hunyuan_vl (#29879) 


Co-authored-by: default avatarIsotr0py <mozf@mail2.sysu.edu.cn>
parent c719c405
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/transformers_utils/processors/hunyuan_vl.py
View file @ 0bec63fa
...@@ -123,7 +123,7 @@ class HunYuanVLProcessor(ProcessorMixin): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment