Update rl.py

This commit is contained in:
Daniel Han 2025-02-19 21:41:17 -08:00
commit 078d623b5d

View file

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