From 5dcc0cd613debcac177f70e86c8f31f956901f87 Mon Sep 17 00:00:00 2001 From: vo1d-ai Date: Fri, 18 Oct 2024 23:46:07 -0400 Subject: [PATCH] fix: compute_loss bug (#1151) Currently, Unsloth doesn't pass additional parameters to Trainer.compute_loss such as return_outputs. This leads to errors when calling trainer.evaluate(). This change fixes the bug by properly passing parameters to Trainer.compute_loss. --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index ea6647594d..0c9468ce9d 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1166,7 +1166,7 @@ def _unsloth_pre_compute_loss(self, model, inputs, *args, **kwargs): inputs["num_items_in_batch"] = kwargs["num_items_in_batch"] pass pass - return self._old_compute_loss(model, inputs, args, kwargs) + return self._old_compute_loss(model, inputs, *args, **kwargs) pass