Add auto_map
Browse files- config.json +12 -4
config.json
CHANGED
@@ -1,4 +1,12 @@
|
|
1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
"encoder": {
|
3 |
"feat_in": 64,
|
4 |
"n_layers": 16,
|
@@ -10,11 +18,11 @@
|
|
10 |
"n_heads": 16,
|
11 |
"conv_kernel_size": 31,
|
12 |
"flash_attn": false
|
13 |
-
|
14 |
"head": {
|
15 |
"feat_in": 768,
|
16 |
"num_classes": 34
|
17 |
-
|
18 |
"labels": [
|
19 |
" ",
|
20 |
"а",
|
@@ -49,6 +57,6 @@
|
|
49 |
"э",
|
50 |
"ю",
|
51 |
"я"
|
52 |
-
|
53 |
-
|
54 |
}
|
|
|
1 |
{
|
2 |
+
"auto_map":{
|
3 |
+
"AutoConfig": "gigaam_transformers.GigaAMConfig",
|
4 |
+
"AutoModel": "gigaam_transformers.GigaAMCTCHF",
|
5 |
+
"AutoProcessor": "gigaam_transformers.GigaAMProcessor",
|
6 |
+
"AutoTokenizer": "gigaam_transformers.GigaAMCTCTokenizer",
|
7 |
+
"AutoFeatureExtractor": "gigaam_transformers.GigaAMFeatureExtractor"
|
8 |
+
},
|
9 |
+
|
10 |
"encoder": {
|
11 |
"feat_in": 64,
|
12 |
"n_layers": 16,
|
|
|
18 |
"n_heads": 16,
|
19 |
"conv_kernel_size": 31,
|
20 |
"flash_attn": false
|
21 |
+
},
|
22 |
"head": {
|
23 |
"feat_in": 768,
|
24 |
"num_classes": 34
|
25 |
+
},
|
26 |
"labels": [
|
27 |
" ",
|
28 |
"а",
|
|
|
57 |
"э",
|
58 |
"ю",
|
59 |
"я"
|
60 |
+
],
|
61 |
+
"blank_id": 33
|
62 |
}
|