diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 5b9aec652a..cd0bb0b391 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -377,6 +377,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): # Patch all functions for function in changed: exec(changed[function], imported_functions, globals()) + print(changed[function]) exec(f"trl.trainer.{trainer_file}.{RLTrainer_name}.{function} = _unsloth_{function}", imported_functions, globals()) pass pass