From 9260ce5721c048d14806e248bfbebb4dcff8497d Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 20 Jun 2025 15:59:09 +0700 Subject: [PATCH] fix: only warn on more than 1 image --- src/axolotl/processing_strategies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/processing_strategies.py b/src/axolotl/processing_strategies.py index ce9b6a838..080697400 100644 --- a/src/axolotl/processing_strategies.py +++ b/src/axolotl/processing_strategies.py @@ -142,7 +142,7 @@ class ProcessingStrategy: # 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 # 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( 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."