fix PaddlePaddle#2665

FrostML · Jul 26, 2022 · 653b276 · 653b276
1 parent 0722c85
commit 653b276
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/transformers/model_outputs.py b/paddlenlp/transformers/model_outputs.py
@@ -91,7 +91,7 @@ def _transformer_encoder_fwd(self,
         if output_attentions:
             all_attentions.append(outputs[-1])
         if cache is not None:
-            new_caches.append(outputs[1])
+            new_caches.append(outputs[0])
 
     if self.norm is not None:
         output = self.norm(output)