tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix: mllama missing aspect ratio

Browse Source
This commit is contained in:
NanoCode012
2025-07-22 10:36:07 +07:00
parent 78169cd37a
commit ee19007ba4
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/utils/collators/mm_chat.py
View File

@@ -92,6 +92,9 @@ class MultiModalChatDataCollator(DataCollatorMixin):
if "pixel_values" in batch: if "pixel_values" in batch:
final_batch["pixel_values"] = torch.stack(batch["pixel_values"]) final_batch["pixel_values"] = torch.stack(batch["pixel_values"])
if "aspect_ratio_ids" in batch:
final_batch["aspect_ratio_ids"] = torch.stack(batch["aspect_ratio_ids"])
if "input_features" in batch: if "input_features" in batch:
final_batch["input_features"] = torch.stack(batch["input_features"]) final_batch["input_features"] = torch.stack(batch["input_features"])