Update config.json
#8
by
Muennighoff
- opened
- config.json +1 -0
config.json
CHANGED
|
@@ -4,6 +4,7 @@
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_rwkv5.Rwkv5Config",
|
|
|
|
| 7 |
"AutoModelForCausalLM": "modeling_rwkv5.Rwkv5ForCausalLM"
|
| 8 |
},
|
| 9 |
"attention_hidden_size": 4096,
|
|
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_rwkv5.Rwkv5Config",
|
| 7 |
+
"AutoModel": "modeling_rwkv5.Rwkv5Model",
|
| 8 |
"AutoModelForCausalLM": "modeling_rwkv5.Rwkv5ForCausalLM"
|
| 9 |
},
|
| 10 |
"attention_hidden_size": 4096,
|