From 722c4ecca63d30f18d2d369d7ba364b12f178ecb Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 11 Feb 2025 16:03:33 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index a5db30d7c3..f7265cff83 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -245,6 +245,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): extra_args += \ "from unsloth_zoo.logging_utils import PatchRLStatistics\n"\ f"PatchRLStatistics('{trainer_file}')\n" + "print(args)\n" # Create RLTrainer args extra_args = extra_args.split("\n")