diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index da163753a4..be9732db34 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -369,10 +369,9 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): ) return try: - RLConfig = eval(f"trl.trainer.{trainer_file}.{RLConfig_name}") - except: + except Exception as e: print( - f"Unsloth: Could not load {RLConfig_name} from trl.trainer.{trainer_file}" + f"Unsloth: Could not load {RLConfig_name} from trl.trainer.{trainer_file}: {e}" ) return