SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add decoder specific error message for T5Stack.forward (#4128)

Browse Source
This commit is contained in:
Lorenzo Ampil
2020-05-03 18:40:08 +08:00
committed by GitHub
parent 1cdd2ad2af
commit 6af3306a1d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/modeling_t5.py
View File

@@ -666,7 +666,10 @@ class T5Stack(T5PreTrainedModel):
elif inputs_embeds is not None: elif inputs_embeds is not None:
input_shape = inputs_embeds.size()[:-1] input_shape = inputs_embeds.size()[:-1]
else: 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: if inputs_embeds is None:
assert self.embed_tokens is not None, "You have to intialize the model with valid token embeddings" assert self.embed_tokens is not None, "You have to intialize the model with valid token embeddings"