Update rl_replacements.py (#3990)
This commit is contained in:
parent
64a9033539
commit
a50f74faa8
1 changed files with 1 additions and 1 deletions
|
|
@ -459,7 +459,7 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
|
|||
|
||||
function = function.replace(string_to_find, replacement_string)
|
||||
|
||||
if trl_version >= Version("0.25.0"):
|
||||
if trl_version >= Version("0.24.0"):
|
||||
# We replace the call using 'completions' with one using 'completions_text'
|
||||
string_to_find = " rewards_per_func = self._calculate_rewards(inputs, prompts, completions, completion_ids_list)"
|
||||
replacement_string = (
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue