Fix paligemma warning message (huggingface#35486)

hiyouga · web-flow · commit 42865860ec6d · 2025-01-02T11:36:53.000+01:00
fix log input
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -519,7 +519,7 @@ def forward(`
`519`	`519`	`# mask out pad-token-ids in labels for BC`
`520`	`520`	`if labels is not None and self.pad_token_id in labels:`
`521`	`521`	`logger.warning_once(`
`522`		- "`labels` contains `pad_token_id` which will be masked with `config.ignore_index`. ",
	`522`	+ "`labels` contains `pad_token_id` which will be masked with `config.ignore_index`. "
`523`	`523`	"You have to mask out `pad_token_id` when preparing `labels`, this behavior will be removed in v.4.46.",
`524`	`524`	`)`
`525`	`525`	`labels = torch.where(input_ids == self.pad_token_id, self.config.ignore_index, labels)`