Update rl.py
This commit is contained in:
parent
381b025779
commit
b15b98c697
1 changed files with 1 additions and 1 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue