Skip to content

Commit 827de30

Browse files
committed
fix bug
1 parent 69c5082 commit 827de30

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddlenlp/transformers/model_outputs.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -209,10 +209,10 @@ def _transformer_decoder_fwd(self,
209209
for i, mod in enumerate(self.layers):
210210
if cache is None:
211211
if self.enable_recompute:
212-
output = recompute(mod, tgt, memory, tgt_mask, memory_mask,
213-
None, output_attentions)
212+
outputs = recompute(mod, tgt, memory, tgt_mask, memory_mask,
213+
None, output_attentions)
214214
else:
215-
output = mod(
215+
outputs = mod(
216216
tgt,
217217
memory,
218218
tgt_mask=tgt_mask,

0 commit comments

Comments
 (0)