diff --git a/src/transformers/models/paligemma/modeling_paligemma.py b/src/transformers/models/paligemma/modeling_paligemma.py index b4a231561ba..9d58fe7eb1b 100644 --- a/src/transformers/models/paligemma/modeling_paligemma.py +++ b/src/transformers/models/paligemma/modeling_paligemma.py @@ -519,7 +519,7 @@ def forward( # mask out pad-token-ids in labels for BC if labels is not None and self.pad_token_id in labels: logger.warning_once( - "`labels` contains `pad_token_id` which will be masked with `config.ignore_index`. ", + "`labels` contains `pad_token_id` which will be masked with `config.ignore_index`. " "You have to mask out `pad_token_id` when preparing `labels`, this behavior will be removed in v.4.46.", ) labels = torch.where(input_ids == self.pad_token_id, self.config.ignore_index, labels)