fix encoder outputs (#8368)

huggingface · Nov 6, 2020 · 0770879 · 0770879
1 parent bc0d26d
commit 0770879
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]