Update config.json
Browse files- config.json +3 -3
config.json
CHANGED
|
@@ -1,10 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
"activation_fn": "swish",
|
| 4 |
-
"architectures": ["
|
| 5 |
"auto_map": {
|
| 6 |
-
"AutoConfig": "
|
| 7 |
-
"AutoModelForCausalLM": "modeling_retnet.
|
| 8 |
},
|
| 9 |
"decoder_embed_dim": 1280,
|
| 10 |
"decoder_ffn_embed_dim": 2560,
|
|
|
|
| 1 |
{
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
"activation_fn": "swish",
|
| 4 |
+
"architectures": ["RetNetForCausalLM"],
|
| 5 |
"auto_map": {
|
| 6 |
+
"AutoConfig": "configuration_retnet.RetNetConfig",
|
| 7 |
+
"AutoModelForCausalLM": "modeling_retnet.RetNetForCausalLM"
|
| 8 |
},
|
| 9 |
"decoder_embed_dim": 1280,
|
| 10 |
"decoder_ffn_embed_dim": 2560,
|