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"}
|