mirror of
https://github.com/unslothai/unsloth
synced 2026-04-21 13:37:39 +00:00
Update rl.py
This commit is contained in:
parent
bae1d69611
commit
0e51ebdd58
1 changed files with 1 additions and 0 deletions
|
|
@ -401,6 +401,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
|
|||
RLTrainer_source,
|
||||
f"trl.trainer.{trainer_file}",
|
||||
imports,
|
||||
overwrite = False,
|
||||
)
|
||||
|
||||
# Patch Trainer
|
||||
|
|
|
|||
Loading…
Reference in a new issue