Update rl.py

This commit is contained in:
Daniel Han 2025-02-13 16:27:23 -08:00
commit 680384fa57

View file

@ -402,7 +402,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
RLTrainer_source,
f"trl.trainer.{trainer_file}",
imports,
overwrite = False,
overwrite = True,
)
# Patch Trainer