423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source 
Former-commit-id: 6e19e56000dd18d5faf84ceabce8d7708ff21e4d
This commit is contained in:
hiyouga 2024-09-01 21:07:49 +08:00
parent 7e4c5d4bb3
commit 559b84dceb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/collator.py
View File

@ -93,7 +93,7 @@ class MultiModalDataCollatorForSeq2Seq(DataCollatorForSeq2Seq):
else: else:
image_grid_thw.append(torch.tensor(feature["image_grid_thw"], dtype=torch.long)) image_grid_thw.append(torch.tensor(feature["image_grid_thw"], dtype=torch.long))
extra_features["image_grid_thw"] = torch.cat(pixel_values, dim=0) extra_features["image_grid_thw"] = torch.cat(image_grid_thw, dim=0)
if extra_features["image_grid_thw"].numel() == 0: if extra_features["image_grid_thw"].numel() == 0:
extra_features["image_grid_thw"] = None extra_features["image_grid_thw"] = None