Update grpo.py (#171)

huggingface · Feb 5, 2025 · 736b59f · 736b59f
1 parent 3fd56dc
commit 736b59f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/open_r1/grpo.py b/src/open_r1/grpo.py
@@ -216,9 +216,7 @@ def make_conversation(example):
 
     # 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: