Skip to content

Commit 0bec63f

Browse files
wkcnIsotr0py
andauthored
[BugFix] fix imgs_pos in hunyuan_vl (vllm-project#29879)
Co-authored-by: Isotr0py <[email protected]>
1 parent c719c40 commit 0bec63f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/transformers_utils/processors/hunyuan_vl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def __call__(
123123

124124
attention_mask = input_ids.ne(self.pad_id)
125125
text_inputs["attention_mask"] = attention_mask
126-
text_inputs["imgs_pos"] = [self.get_imgs_pos(input_ids)]
126+
text_inputs["imgs_pos"] = [self.get_imgs_pos(e) for e in input_ids]
127127
# image_inputs["imgs"] = [[image_inputs["pixel_values"]]]
128128

129129
return_tensors = kwargs.pop("return_tensors", None)

0 commit comments

Comments
 (0)