fix encoder outputs (#8368)

2020-11-06 21:03:25 +01:00
parent bc0d26d1de
commit 07708793f2
1 changed files with 1 additions and 2 deletions
--- a/src/transformers/generation_tf_utils.py
+++ b/src/transformers/generation_tf_utils.py
@@ -348,8 +348,7 @@ class TFGenerationMixin:
                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]