fix conditional for None values

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

View File

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

View File