Update rl.py

This commit is contained in:
Daniel Han 2025-02-19 17:58:25 -08:00
commit 8368734e48

View file

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