diff --git a/paddleformers/transformers/glm4_moe/modeling.py b/paddleformers/transformers/glm4_moe/modeling.py index 7e1a7c02a4b..ea28a4c9c94 100644 --- a/paddleformers/transformers/glm4_moe/modeling.py +++ b/paddleformers/transformers/glm4_moe/modeling.py @@ -88,6 +88,9 @@ class GLMMoEModelProvider(GPTModelProvider): bias_dropout_fusion: bool = True moe_expert_fusion: bool = False + attention_softmax_in_fp32: bool = True + bf16: bool = True + def eager_attention_forward( module: nn.Layer,