From ee19007ba4a51af406a9e80d2b1b53ffc7d713a7 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Tue, 22 Jul 2025 10:36:07 +0700 Subject: [PATCH] fix: mllama missing aspect ratio --- src/axolotl/utils/collators/mm_chat.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/axolotl/utils/collators/mm_chat.py b/src/axolotl/utils/collators/mm_chat.py index dba27f7d8..c5f6a62e3 100644 --- a/src/axolotl/utils/collators/mm_chat.py +++ b/src/axolotl/utils/collators/mm_chat.py @@ -92,6 +92,9 @@ class MultiModalChatDataCollator(DataCollatorMixin): if "pixel_values" in batch: 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: final_batch["input_features"] = torch.stack(batch["input_features"])