diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index c6ec59132..2b467f2b7 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -912,7 +912,6 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): kwargs["return_tensors"] = "pt" if issubclass(collator, DataCollatorForSeq2Seq): - print(type(collator), issubclass(collator, DataCollatorForSeq2Seq)) kwargs["sequence_parallel_degree"] = training_args.sequence_parallel_degree return collator(