Update modeling_minGRULM.py

modeling_minGRULM.py

@@ -127,19 +127,18 @@ class MinGRULMForCausalLM(PreTrainedModel):
         This includes the wrapped model and any extra components like the language model head.
         """
         state_dict = {}
-        
+    
         # Add min_gru_model's state_dict
         state_dict['model'] = self.model.min_gru_model.state_dict()
         
         # Add lm_head's state_dict
         state_dict['lm_head'] = self.lm_head.state_dict()
-        
-        # Optionally, add config if needed
-        state_dict['config'] = self.config.state_dict()
     
+        # Add config as a dictionary (not state_dict, since it is not available)
+        state_dict['config'] = self.config.__dict__
+        
         return state_dict
 
-
     @classmethod
     def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
         """