Commit
•
71b76e3
1
Parent(s):
0cd567a
Add "archetectures" key to config.json (#4)
Browse files- Add "archetectures" key to config.json (0a48d458112a6e3bbd4d0ebfda651fd27ecf5025)
Co-authored-by: Christopher Atanasopulo <Christ0pher@users.noreply.huggingface.co>
- config.json +4 -1
config.json
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
| 2 |
"vocab_size": 50257,
|
| 3 |
"hidden_size": 4096,
|
| 4 |
"tie_word_embeddings": true,
|
|
@@ -8,4 +11,4 @@
|
|
| 8 |
"n_head": 32,
|
| 9 |
"attention_dropout": 0.0,
|
| 10 |
"model_type": "bloom"
|
| 11 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"BloomForCausalLM"
|
| 4 |
+
],
|
| 5 |
"vocab_size": 50257,
|
| 6 |
"hidden_size": 4096,
|
| 7 |
"tie_word_embeddings": true,
|
|
|
|
| 11 |
"n_head": 32,
|
| 12 |
"attention_dropout": 0.0,
|
| 13 |
"model_type": "bloom"
|
| 14 |
+
}
|