fix script

huggingface · Jan 23, 2025 · d6e3fbf · d6e3fbf
1 parent fc7a85b
commit d6e3fbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/scripts/grpo.py b/trl/scripts/grpo.py
@@ -60,7 +60,7 @@ def main(script_args, training_args, model_args):
     # Initialize the GRPO trainer
     trainer = GRPOTrainer(
         model=model,
-        reward_model=reward_model,
+        reward_funcs=reward_model,
         args=training_args,
         train_dataset=dataset[script_args.dataset_train_split],
         eval_dataset=dataset[script_args.dataset_test_split] if training_args.eval_strategy != "no" else None,