Skip to content

Commit 595f386

Browse files
committed
Fix the call of assign in nlp samples.
1 parent 37b1b2f commit 595f386

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle_samples/PaddleNLP/mbart-large-cc25/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -526,7 +526,7 @@ def forward(
526526
data_1,
527527
):
528528
# pd_op.assign: (1x22xi64) <- (1x22xi64)
529-
assign_0 = data_0
529+
assign_0 = paddle._C_ops.assign(data_0)
530530

531531
# pd_op.flatten: (22xi64) <- (1x22xi64)
532532
flatten_0 = paddle._C_ops.flatten(assign_0, 0, 1)

paddle_samples/PaddleNLP/mbart-large-en-ro/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -526,7 +526,7 @@ def forward(
526526
data_1,
527527
):
528528
# pd_op.assign: (1x22xi64) <- (1x22xi64)
529-
assign_0 = data_0
529+
assign_0 = paddle._C_ops.assign(data_0)
530530

531531
# pd_op.flatten: (22xi64) <- (1x22xi64)
532532
flatten_0 = paddle._C_ops.flatten(assign_0, 0, 1)

0 commit comments

Comments
 (0)