From f6b4b5754b6aaa8b549afae3e3464652cd6b0036 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Fri, 6 Dec 2024 13:11:58 +0530 Subject: [PATCH] nit --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index af908e48e4b8..8b9e14aa6fe0 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -3649,7 +3649,7 @@ def training_step( return loss_mb.reduce_mean().detach().to(self.args.device) with self.compute_loss_context_manager(): - if self.model_accepts_loss_kwargs: + if not self.model_accepts_loss_kwargs: loss = self.compute_loss(model, inputs) else: loss = self.compute_loss(model, inputs, num_items_in_batch=num_items_in_batch)