Update t5.rst (huggingface#14776)

Xing Han Lu · web-flow · commit 72c6e8b8bf89 · 2021-12-15T14:59:11.000+01:00
diff --git a/docs/source/model_doc/t5.rst b/docs/source/model_doc/t5.rst
@@ -191,10 +191,8 @@ language modeling head on top of the decoder.
         labels = target_encoding.input_ids
 
         # replace padding token id's of the labels by -100
-        labels = [
-                   [(label if label != tokenizer.pad_token_id else -100) for label in labels_example] for labels_example in labels
-        ] 
         labels = torch.tensor(labels)
+        labels[labels == tokenizer.pad_token_id] = -100
 
         # forward pass
         loss = model(input_ids=input_ids, attention_mask=attention_mask, labels=labels).loss
