Update rl.py

This commit is contained in:
Daniel Han 2025-02-11 23:47:33 -08:00
parent bae1d69611
commit 0e51ebdd58

View file

@ -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