fix rms norm when dy2statuc (#58868)

PaddlePaddle · Nov 10, 2023 · 0a3b495 · 0a3b495
1 parent f55021f
commit 0a3b495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/incubate/nn/functional/fused_rms_norm.py b/python/paddle/incubate/nn/functional/fused_rms_norm.py
@@ -106,7 +106,7 @@ def fused_rms_norm(
     outputs_dict['residual_out'] = residual_out
 
     inputs = {'x': x, 'norm_weight': norm_weight}
-    if norm_bias:
+    if norm_bias is not None:
         inputs['norm_bias'] = norm_bias
     if residual is not None:
         inputs['residual'] = residual