Update rl.py

This commit is contained in:
Daniel Han 2025-02-11 00:37:08 -08:00
commit 322225ce70

View file

@ -295,6 +295,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
RLTrainer_extras = RLTrainer_extras,
)
print(RLTrainer_source)
# Create new function
created_module = create_new_function(