Adding config_class for FlexBertForMaskedLM

modeling_flexbert.py

@@ -1005,6 +1005,7 @@ class FlexBertModel(FlexBertPreTrainedModel):
 
 
 class FlexBertForMaskedLM(FlexBertPreTrainedModel):
+    config_class = FlexBertConfig
     def __init__(self, config: FlexBertConfig):
         super().__init__(config)
         self.bert = FlexBertModel(config)