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

03885a3f · Ayal Klein · GitHub · 501307b5 · 03885a3f
Unverified Commit 03885a3f authored Dec 27, 2021 by Ayal Klein Committed by GitHub Dec 27, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/transformers/data/data_collator.py src/transformers/data/data_collator.py +5 -1

No files found.
--- a/src/transformers/data/data_collator.py
+++ b/src/transformers/data/data_collator.py
@@ -591,7 +591,11 @@ class DataCollatorForSeq2Seq:
        )
        # 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