Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
@@ -6,8 +6,8 @@
|
|
6 |
"RetNetForCausalLM"
|
7 |
],
|
8 |
"auto_map": {
|
9 |
-
"AutoConfig": "
|
10 |
-
"AutoModelForCausalLM": "
|
11 |
},
|
12 |
"decoder_embed_dim": 1024,
|
13 |
"decoder_ffn_embed_dim": 1728,
|
|
|
6 |
"RetNetForCausalLM"
|
7 |
],
|
8 |
"auto_map": {
|
9 |
+
"AutoConfig": "isek-ai/SDPrompt-RetNet-v2--configuration_retnet.RetNetConfig",
|
10 |
+
"AutoModelForCausalLM": "isek-ai/SDPrompt-RetNet-v2--modeling_retnet.RetNetForCausalLM"
|
11 |
},
|
12 |
"decoder_embed_dim": 1024,
|
13 |
"decoder_ffn_embed_dim": 1728,
|