fix to issue huggingface#14833 in data_collator - consider no labels (h…

…uggingface#14930)
stevhliu · Jan 6, 2022 · a439438 · a439438
1 parent 1000054
commit a439438
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py
@@ -591,7 +591,11 @@ def __call__(self, features, return_tensors=None):
         )
 
         # prepare decoder_input_ids
-        if self.model is not None and hasattr(self.model, "prepare_decoder_input_ids_from_labels"):
+        if (
+            labels is not None
+            and self.model is not None
+            and hasattr(self.model, "prepare_decoder_input_ids_from_labels")
+        ):
             decoder_input_ids = self.model.prepare_decoder_input_ids_from_labels(labels=features["labels"])
             features["decoder_input_ids"] = decoder_input_ids