fix out order

PaddlePaddle · Feb 17, 2024 · bd88caa · bd88caa
1 parent bb0fa3a
commit bd88caa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/incubate/nn/functional/fused_transformer.py b/python/paddle/incubate/nn/functional/fused_transformer.py
@@ -1152,7 +1152,7 @@ def fused_multi_transformer(
     )  # semantic transfer
 
     if in_dynamic_mode():
-        cache_kv_out, final_out = _C_ops.fused_multi_transformer(
+        final_out, cache_kv_out = _C_ops.fused_multi_transformer(
             x,
             ln_scales,
             ln_biases,