From 4b709502fd958b246ae3b20de1610fb4f153af91 Mon Sep 17 00:00:00 2001 From: Yacklin Wong Date: Mon, 3 Nov 2025 20:01:12 -0400 Subject: [PATCH] Correct syntax error in trainer.md A comma is missing between two parameters in the signature of compute_loss function. --- docs/source/en/trainer.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/trainer.md b/docs/source/en/trainer.md index 045f0837c334..1d700d398b5c 100644 --- a/docs/source/en/trainer.md +++ b/docs/source/en/trainer.md @@ -187,7 +187,7 @@ from torch import nn from transformers import Trainer class CustomTrainer(Trainer): - def compute_loss(self, model: nn.Module, inputs: dict[str, Union[torch.Tensor, Any]], return_outputs: bool = False num_items_in_batch: Optional[torch.Tensor] = None): + def compute_loss(self, model: nn.Module, inputs: dict[str, Union[torch.Tensor, Any]], return_outputs: bool = False, num_items_in_batch: Optional[torch.Tensor] = None): labels = inputs.pop("labels") # forward pass outputs = model(**inputs)