update return value

PaddlePaddle · Mar 11, 2022 · 67f3b8e · 67f3b8e · paddle-bot-old · Mar 11, 2022
1 parent c003600
commit 67f3b8e
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
@@ -487,4 +487,4 @@ def fused_multi_head_attention(x,
             },
             attrs=attrs)
 
-        return final_out, cache_kv_out if cache_kv else final_out
+        return (final_out, cache_kv_out) if cache_kv else final_out