sino commited on
Commit
f0c5546
·
1 Parent(s): 92f032f

Update configuration_maelm.py

Browse files
Files changed (1) hide show
  1. configuration_maelm.py +2 -0
configuration_maelm.py CHANGED
@@ -127,3 +127,5 @@ class MAELMConfig(PretrainedConfig):
127
  self._name_or_path = None
128
  self.resume_from_checkpoint = resume_from_checkpoint
129
  self.resume_from_pth = resume_from_pth
 
 
 
127
  self._name_or_path = None
128
  self.resume_from_checkpoint = resume_from_checkpoint
129
  self.resume_from_pth = resume_from_pth
130
+ self.auto_map = {"AutoConfig": "configuration_maelm.MAELMConfig",
131
+ "AutoModel": "modeling_maelm.MAEForCausalLM"}