From 03885a3f505a5524f85e3391bfd946a681c98d36 Mon Sep 17 00:00:00 2001 From: Ayal Klein Date: Mon, 27 Dec 2021 18:48:48 +0200 Subject: [PATCH] fix to issue #14833 in data_collator - consider no labels (#14930) --- src/transformers/data/data_collator.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py index f6750268cb..beeca57650 100644 --- 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