Skip to content

Commit 75e51df

Browse files
authored
fix bug (#10862)
1 parent 33e01f1 commit 75e51df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/transformers/deepseek_v2/modeling_pp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -435,7 +435,7 @@ def __init__(self, forward_nodes, backward_nodes, use_fuion=True):
435435
def forward_backward(self, inputs, output_grad, event_to_wait=None):
436436
for n in self.nodes:
437437
inputs, output_grad, event_to_wait = n.forward_backward(inputs, output_grad, event_to_wait)
438-
return inputs, output_grad
438+
return inputs, output_grad, None
439439

440440

441441
class OverlapedScheduleNode:

0 commit comments

Comments
 (0)