Update unsloth/models/rl.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Daniel Han 2025-12-01 07:19:50 -08:00 committed by GitHub
commit e38209de83

View file

@ -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