Update rl.py
This commit is contained in:
parent
6be29a36a6
commit
97e3f9e69b
1 changed files with 1 additions and 1 deletions
|
|
@ -672,7 +672,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
|
|||
RLTrainer_source,
|
||||
f"trl.trainer.{trainer_file}",
|
||||
imports,
|
||||
overwrite = False,
|
||||
overwrite = True,
|
||||
)
|
||||
|
||||
# Patch Trainer
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue