fix rms norm when dy2statuc (PaddlePaddle#58868)

danleifeng · Nov 10, 2023 · 034ae26 · 034ae26
1 parent ff72c36
commit 034ae26
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