[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
8b5130ae2d
commit
806f8d2d7e
1 changed files with 4 additions and 4 deletions
|
|
@ -388,12 +388,12 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
|
|||
patched = patched[: match.start()] + wrapped + patched[match.end() :]
|
||||
|
||||
function = patched
|
||||
|
||||
|
||||
function = function.replace(
|
||||
" return output",
|
||||
""" if not _was_training:
|
||||
" return output",
|
||||
""" if not _was_training:
|
||||
self.model.for_inference()
|
||||
return output"""
|
||||
return output""",
|
||||
)
|
||||
|
||||
return function
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue