fix #2665 (#2877)

PaddlePaddle · Jul 26, 2022 · 8609c92 · 8609c92
1 parent 174ce23
commit 8609c92
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)