fix to issue #14833 in data_collator - consider no labels (#14930)

huggingface · Dec 27, 2021 · 03885a3 · 03885a3
1 parent 501307b
commit 03885a3
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