From 6af3306a1da0322f58861b1fbb62ce5223d97b8a Mon Sep 17 00:00:00 2001 From: Lorenzo Ampil Date: Sun, 3 May 2020 18:40:08 +0800 Subject: [PATCH] Add decoder specific error message for T5Stack.forward (#4128) --- src/transformers/modeling_t5.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/transformers/modeling_t5.py b/src/transformers/modeling_t5.py index 73df985e31..d8d61e0f02 100644 --- a/src/transformers/modeling_t5.py +++ b/src/transformers/modeling_t5.py @@ -666,7 +666,10 @@ class T5Stack(T5PreTrainedModel): elif inputs_embeds is not None: input_shape = inputs_embeds.size()[:-1] else: - raise ValueError("You have to specify either input_ids or inputs_embeds") + if self.is_decoder: + raise ValueError("You have to specify either decoder_input_ids or decoder_inputs_embeds") + else: + raise ValueError("You have to specify either input_ids or inputs_embeds") if inputs_embeds is None: assert self.embed_tokens is not None, "You have to intialize the model with valid token embeddings"