Update rl.py

This commit is contained in:
Daniel Han 2025-02-19 23:15:13 -08:00
commit b15b98c697

View file

@ -450,7 +450,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
pass
# Check if SamplingParams is in there
if "SamplingParams" in RLTrainer_source:
if "SamplingParams" in old_RLTrainer_source:
RL_pre = RL_pre + "\n" + inspect.getsource(vLLMSamplingParams)
pass