From d2c32d0cba8175738a0c74af901086e9647ae7ba Mon Sep 17 00:00:00 2001 From: bursteratom Date: Fri, 6 Dec 2024 13:04:42 -0500 Subject: [PATCH] lint --- src/axolotl/utils/collators/mm_chat.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/axolotl/utils/collators/mm_chat.py b/src/axolotl/utils/collators/mm_chat.py index 7ec9c77f9..799428088 100644 --- a/src/axolotl/utils/collators/mm_chat.py +++ b/src/axolotl/utils/collators/mm_chat.py @@ -213,11 +213,9 @@ class MultiModalChatDataCollator(DataCollatorMixin): for example in examples ] - if chat_template_type != "llava": - images = __class__.process_images(examples, max_images=max_images) - else: + images = __class__.process_images(examples, max_images=max_images) + if chat_template_type == "llava": # LLava1.5 does not support multiple images - images = [example["images"] for example in examples] images = [image[0] for image in images] # Tokenize the texts and process the images