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

fix conditional for None values

Browse Source
This commit is contained in:
Wing Lian
2025-08-17 12:49:48 -04:00
parent 7fd3d8abc4
commit bb65157dcf
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/builders/causal.py
View File

@@ -424,7 +424,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
): ):
if training_args.pretraining: if training_args.pretraining:
if ( if (
self.cfg.pretraining_sample_concatenation is False not self.cfg.pretraining_sample_concatenation
or self.cfg.micro_batch_size > 1 or self.cfg.micro_batch_size > 1
): ):
return DataCollatorForSeq2Seq(self.tokenizer, **kwargs) return DataCollatorForSeq2Seq(self.tokenizer, **kwargs)

0
src/axolotl/exception_handling.py Normal file
View File