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-13 17:15:29 -08:00
commit b53617e6cf
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/rl_replacements.py
View file

@ -105,13 +105,13 @@ def sft_trainer_compute_loss(function_name, function):
if function_name != "compute_loss": return function
def compute_loss(self, model, inputs, return_outputs = False, num_items_in_batch = None):
(loss, outputs) = super().compute_loss(
outputs = super().compute_loss(
model,
inputs,
return_outputs = return_outputs,
num_items_in_batch = num_items_in_batch,
)
return (loss, outputs) if return_outputs else loss
return outputs
pass
function = inspect.getsource(compute_loss)