don't automatically ifer reporters

neuralmagic · May 3, 2024 · 205c5fc · 205c5fc
1 parent 54a2cde
commit 205c5fc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/sparseml/transformers/finetune/text_generation.py b/src/sparseml/transformers/finetune/text_generation.py
@@ -85,8 +85,12 @@ def apply(**kwargs):
     """
     CLI entrypoint for any of training, eval, predict or oneshot
     """
+    report_to = kwargs.get("report_to", None)
     model_args, data_args, training_args = parse_args(**kwargs)
     training_args.run_stages = True
+    if report_to is None:  # user didn't specify any reporters
+        # get rid of the reporters inferred from hugging face
+        training_args.report_to = []
     main(model_args, data_args, training_args)