fix encoder outputs (#8368)

07708793 · Patrick von Platen · GitHub · bc0d26d1 · 07708793
Unverified Commit 07708793 authored Nov 06, 2020 by Patrick von Platen Committed by GitHub Nov 06, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

src/transformers/generation_tf_utils.py src/transformers/generation_tf_utils.py +1 -2

No files found.
--- 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]