fix

modelscope · Dec 24, 2024 · f0195d0 · f0195d0
1 parent 288df5b
commit f0195d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/trainers/mixin.py b/swift/trainers/mixin.py
@@ -253,7 +253,7 @@ def _patch_loss_function(self):
             yield
             return
 
-        loss_function = model.loss_function
+        loss_function = self.model.loss_function
 
         @warps(loss_function)
         def new_loss_function(logits, labels, **kwargs):