From b809d2f073482087345596eecfe36c1159c1763b Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 5 Apr 2020 12:23:09 +0200 Subject: [PATCH] Fix TF T5 docstring (#3636) --- src/transformers/modeling_tf_t5.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/modeling_tf_t5.py b/src/transformers/modeling_tf_t5.py index bfd5a369d1..eede1fd675 100644 --- a/src/transformers/modeling_tf_t5.py +++ b/src/transformers/modeling_tf_t5.py @@ -731,7 +731,7 @@ class TFT5Model(TFT5PreTrainedModel): tokenizer = T5Tokenizer.from_pretrained('t5-small') model = TFT5Model.from_pretrained('t5-small') input_ids = tokenizer.encode("Hello, my dog is cute", return_tensors="tf") # Batch size 1 - outputs = model(input_ids, input_ids=input_ids) + outputs = model(input_ids, decoder_input_ids=input_ids) last_hidden_states = outputs[0] # The last hidden-state is the first element of the output tuple """ @@ -829,7 +829,7 @@ class TFT5ForConditionalGeneration(TFT5PreTrainedModel): tokenizer = T5Tokenizer.from_pretrained('t5-small') model = TFT5ForConditionalGeneration.from_pretrained('t5-small') input_ids = tokenizer.encode("Hello, my dog is cute", return_tensors="tf") # Batch size 1 - outputs = model(input_ids, input_ids=input_ids) + outputs = model(input_ids, decoder_input_ids=input_ids) prediction_scores = outputs[0] tokenizer = T5Tokenizer.from_pretrained('t5-small')