diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 0d795aa282..e1b40e64ed 100644 --- 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}"