File size: 1,640 Bytes
9b9aeff
 
 
 
1b70882
9b9aeff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
caa57df
9b9aeff
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
{   
    "auto_map":{
        "AutoConfig": "gigaam_transformers.GigaAMConfig",
        "AutoModel": "gigaam_transformers.GigaAMRNNTHF",
        "AutoModelForSpeechSeq2Seq": "gigaam_transformers.GigaAMRNNTHF",
        "AutoProcessor": "gigaam_transformers.GigaAMProcessor",
        "AutoTokenizer": "gigaam_transformers.GigaAMTokenizer",
        "AutoFeatureExtractor": "gigaam_transformers.GigaAMFeatureExtractor"
    },

    "encoder": {
        "feat_in": 64,
        "n_layers": 16,
        "d_model": 768,
        "subsampling_factor": 4,
        "ff_expansion_factor": 4,
        "self_attention_model": "rotary",
        "pos_emb_max_len": 5000,
        "n_heads": 16,
        "conv_kernel_size": 31,
        "flash_attn": false
    },
    "head": {
        "decoder": {
            "pred_hidden": 320,
            "pred_rnn_layers": 1,
            "num_classes": 34
            },
        "joint": {
            "enc_hidden": 768,
            "pred_hidden": 320,
            "joint_hidden": 320,
            "num_classes": 34
            }
    },
    "labels": [
        " ",
        "а",
        "б",
        "в",
        "г",
        "д",
        "е",
        "ж",
        "з",
        "и",
        "й",
        "к",
        "л",
        "м",
        "н",
        "о",
        "п",
        "р",
        "с",
        "т",
        "у",
        "ф",
        "х",
        "ц",
        "ч",
        "ш",
        "щ",
        "ъ",
        "ы",
        "ь",
        "э",
        "ю",
        "я"
    ],
    "blank_id": 33,
    "pad_token_id": 33,
    "max_symbols": 1000,
    "model_type": "gigaam-rnnt"
}