Update rl.py

2025-02-17 19:58:17 -08:00 · 2025-02-17 19:58:17 -08:00 · 8bdcfec233
commit 8bdcfec233
parent abc16ad45f
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@ -429,6 +429,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
    # Get all pre-modules
    if trainer_file in RL_PRE_ITEMS:
        RL_pre = "\n".join(RL_PRE_ITEMS[trainer_file])
+        print(RL_pre)
    else:
        RL_pre = ""
    pass