fix encoder outputs (huggingface#8368)

fabiocapsouza · Nov 15, 2020 · 54d6a69 · 54d6a69
1 parent b390ebd
commit 54d6a69
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/transformers/generation_tf_utils.py b/src/transformers/generation_tf_utils.py
@@ -348,8 +348,7 @@ def generate(
                 shape=(-1,),
             )
             # expand encoder_outputs
-            encoder_outputs = (tf.gather(encoder_outputs[0], expanded_batch_idxs, axis=0), *encoder_outputs[1:])
-
+            encoder_outputs = (tf.gather(encoder_outputs[0], expanded_batch_idxs, axis=0),)
         else:
             encoder_outputs = None
             cur_len = shape_list(input_ids)[-1]