sino
commited on
Commit
·
f0c5546
1
Parent(s):
92f032f
Update configuration_maelm.py
Browse files- 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"}
|