Update sft.py

https://github.com/huggingface/open-r1/pull/171/files
huggingface · Feb 5, 2025 · a14baf6 · a14baf6
1 parent 138df0c
commit a14baf6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/open_r1/sft.py b/src/open_r1/sft.py
@@ -167,9 +167,7 @@ def main(script_args, training_args, model_args):
 
     # Save everything else on main process
     kwargs = {
-        "finetuned_from": model_args.model_name_or_path,
-        "dataset": list(script_args.dataset_name),
-        "dataset_tags": list(script_args.dataset_name),
+        "dataset_name": script_args.dataset_name,
         "tags": ["open-r1"],
     }
     if trainer.accelerator.is_main_process: