Bug fix: use correct mm_items in embed_mm_inputs (#8893)

This commit is contained in:
Binyao Jiang
2025-08-16 19:55:56 -07:00
committed by GitHub
parent 1c1f8a118e
commit 66d6be0874
3 changed files with 62 additions and 1 deletions

View File

@@ -560,7 +560,7 @@ def embed_mm_inputs(
]
items_size[i + 1] = len(mm_items)
items_offsets.append(
flatten_nested_list([item.offsets for item in mm_inputs.mm_items])
flatten_nested_list([item.offsets for item in mm_items])
)
items_size = torch.cumsum(items_size, dim=0).tolist()