From 205c5fc63f405949ba62334cc2e049ea8916466a Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Fri, 3 May 2024 16:05:50 +0000 Subject: [PATCH] don't automatically ifer reporters --- src/sparseml/transformers/finetune/text_generation.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/sparseml/transformers/finetune/text_generation.py b/src/sparseml/transformers/finetune/text_generation.py index fbc1bcb146c..0ade9b9974a 100644 --- 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)