removing print statement

2025-03-17 15:32:28 +00:00
parent 2727d86544
commit a09d1ccbf2
1 changed files with 0 additions and 1 deletions
--- 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(