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

--no-verify

Browse Source 
fixes silly mistake
This commit is contained in:
Sunny Liu
2025-01-30 14:08:26 -05:00
parent 065f6d477e
commit 93a268e43d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/trainer_builder.py
View File

@@ -1956,7 +1956,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
elif use_batch_sampler_collator: elif use_batch_sampler_collator:
if self.cfg.flex_attention is True: if self.cfg.flex_attention is True:
collator = FlexBatchSamplerDataCollatorForSeq2Seq 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 collator = V2BatchSamplerDataCollatorForSeq2Seq
elif ( elif (
self.cfg.model_config_type in ["llama"] self.cfg.model_config_type in ["llama"]