ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update rl_replacements.py

Browse source
This commit is contained in:
Daniel Han 2025-02-11 22:02:41 -08:00
commit c643d950f7
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/rl_replacements.py
View file

@ -110,7 +110,7 @@ def sft_trainer_compute_loss(function_name, function):
)
if len(replacer) != 0:
replacer = replacer[0]
returner = " "*8 + "return (loss, outputs) if return_outputs else loss"
returner = "\n" + " "*8 + "return (loss, outputs) if return_outputs else loss"
function = function.replace(replacer, replacer + returner)
pass
return function