Upload configuration.py
Browse files- configuration.py +2 -1
configuration.py
CHANGED
|
@@ -24,7 +24,8 @@ class MoEGPTConfig(PretrainedConfig):
|
|
| 24 |
architectures=["MoEGPTForCausalLM"],
|
| 25 |
auto_map={
|
| 26 |
"AutoConfig": "configuration.MoEGPTConfig",
|
| 27 |
-
"AutoModelForCausalLM": "modeling.MoEGPTForCausalLM"
|
|
|
|
| 28 |
},
|
| 29 |
**kwargs,
|
| 30 |
):
|
|
|
|
| 24 |
architectures=["MoEGPTForCausalLM"],
|
| 25 |
auto_map={
|
| 26 |
"AutoConfig": "configuration.MoEGPTConfig",
|
| 27 |
+
"AutoModelForCausalLM": "modeling.MoEGPTForCausalLM",
|
| 28 |
+
"AutoTokenizer": "transformers.GPT2TokenizerFast"
|
| 29 |
},
|
| 30 |
**kwargs,
|
| 31 |
):
|