From 251066289ddb429ba69079f0d3da7c7a6acf1cbc Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 28 May 2025 06:04:23 -0700 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 48a8b00481..e5cb226433 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -395,7 +395,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): if trainer_file in RL_METRICS_CHANGES: process_extra_args = RL_METRICS_CHANGES[trainer_file] for process_extra_arg in process_extra_args: - other_metrics_processor += process_extra_arg(call_args, extra_args) + other_metrics_processor += process_extra_arg(old_RLTrainer_source, old_RLConfig_source) pass # Add statistics as well!