From 93a268e43d9b43fbb5781b60eff4f42455f29ddb Mon Sep 17 00:00:00 2001 From: Sunny Liu Date: Thu, 30 Jan 2025 14:08:26 -0500 Subject: [PATCH] --no-verify fixes silly mistake --- src/axolotl/core/trainer_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index 6f3f9f466..38b4c31d8 100755 --- 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"]