--no-verify

fixes silly mistake
2025-01-30 14:08:26 -05:00
parent 065f6d477e
commit 93a268e43d
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/core/trainer_builder.py
+++ b/src/axolotl/core/trainer_builder.py
@@ -1956,7 +1956,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
        elif use_batch_sampler_collator:
            if self.cfg.flex_attention is True:
                collator = FlexBatchSamplerDataCollatorForSeq2Seq
-            if self.cfg.model_config_type in SUPPORTED_MULTIPACK_MODEL_TYPES:
+            elif self.cfg.model_config_type in SUPPORTED_MULTIPACK_MODEL_TYPES:
                collator = V2BatchSamplerDataCollatorForSeq2Seq
            elif (
                self.cfg.model_config_type in ["llama"]