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

lint

Browse Source
This commit is contained in:
bursteratom
2024-12-06 13:04:42 -05:00
parent cec9887609
commit d2c32d0cba
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -213,11 +213,9 @@ class MultiModalChatDataCollator(DataCollatorMixin):
for example in examples for example in examples
] ]
if chat_template_type != "llava": images = __class__.process_images(examples, max_images=max_images)
images = __class__.process_images(examples, max_images=max_images) if chat_template_type == "llava":
else:
# LLava1.5 does not support multiple images # LLava1.5 does not support multiple images
images = [example["images"] for example in examples]
images = [image[0] for image in images] images = [image[0] for image in images]
# Tokenize the texts and process the images # Tokenize the texts and process the images