Update rl.py

2025-12-01 08:00:08 -08:00 · 2025-12-01 08:00:08 -08:00 · d33a5b909c
commit d33a5b909c
parent a00ec814e5
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@ -369,6 +369,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
        )
        return
    try:
+        RLConfig = eval(f"trl.trainer.{trainer_file}.{RLConfig_name}")
    except Exception as e:
        print(
            f"Unsloth: Could not load {RLConfig_name} from trl.trainer.{trainer_file}: {e}"