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

fix: only warn on more than 1 image

Browse Source
This commit is contained in:
NanoCode012
2025-06-20 15:59:09 +07:00
parent b5e6e20e55
commit 9260ce5721
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/processing_strategies.py
View File

@@ -142,7 +142,7 @@ class ProcessingStrategy:
# TODO: check if it's normal to be single image only for common datasets # TODO: check if it's normal to be single image only for common datasets
# From observation, it's usually a list of single image but some datasets may have several columns for images # From observation, it's usually a list of single image but some datasets may have several columns for images
# Temporary solution: take the first image and suggest people convert their datasets to use multi-content Messages # Temporary solution: take the first image and suggest people convert their datasets to use multi-content Messages
if len(processed_example[image_key]) > 0: if len(processed_example[image_key]) > 1:
LOG.warning( LOG.warning(
f"Found {len(processed_example[image_key])} images in a sample. Using the first one." f"Found {len(processed_example[image_key])} images in a sample. Using the first one."
"If you are using a dataset with multiple images per sample, please convert it to use multi-content Messages." "If you are using a dataset with multiple images per sample, please convert it to use multi-content Messages."