Fix architecture (#6)
Browse files- Fix architecture (51803c1b574b867545c7f6c668e9090c025a24cb)
- config.json +2 -2
config.json
CHANGED
@@ -5,7 +5,7 @@
|
|
5 |
"bias_dropout_fusion": true,
|
6 |
"bos_token_id": 1,
|
7 |
"architectures": [
|
8 |
-
"
|
9 |
],
|
10 |
"eos_token_id": 2,
|
11 |
"pad_token_id": 3,
|
@@ -27,4 +27,4 @@
|
|
27 |
"transformers_version": "4.20.0",
|
28 |
"use_cache": true,
|
29 |
"vocab_size": 250880
|
30 |
-
}
|
|
|
5 |
"bias_dropout_fusion": true,
|
6 |
"bos_token_id": 1,
|
7 |
"architectures": [
|
8 |
+
"BloomForCausalLM"
|
9 |
],
|
10 |
"eos_token_id": 2,
|
11 |
"pad_token_id": 3,
|
|
|
27 |
"transformers_version": "4.20.0",
|
28 |
"use_cache": true,
|
29 |
"vocab_size": 250880
|
30 |
+
}
|