From 4c520db499b2591f0f78e2a2b50c17a5a7ff47f6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 16:35:09 -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 b8d191dcfe..fadae874db 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -404,6 +404,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): imports, overwrite = True, ) + print("###") # Patch Trainer exec(f"trl.{RLTrainer_name} = created_module.Unsloth{RLTrainer_name}", locals(), globals())