Jingya HF Staff commited on
Commit
e4305b4
·
verified ·
1 Parent(s): 5003912

Upload encoder/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. encoder/config.json +187 -0
encoder/config.json ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "_name_or_path": "openai/whisper-tiny",
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "apply_spec_augment": false,
7
+ "architectures": [
8
+ "WhisperForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.0,
11
+ "begin_suppress_tokens": [
12
+ 220,
13
+ 50257
14
+ ],
15
+ "bos_token_id": 50257,
16
+ "classifier_proj_size": 256,
17
+ "d_model": 384,
18
+ "decoder_attention_heads": 6,
19
+ "decoder_ffn_dim": 1536,
20
+ "decoder_layerdrop": 0.0,
21
+ "decoder_layers": 4,
22
+ "decoder_start_token_id": 50258,
23
+ "dropout": 0.0,
24
+ "encoder_attention_heads": 6,
25
+ "encoder_ffn_dim": 1536,
26
+ "encoder_layerdrop": 0.0,
27
+ "encoder_layers": 4,
28
+ "eos_token_id": 50257,
29
+ "forced_decoder_ids": [
30
+ [
31
+ 1,
32
+ 50259
33
+ ],
34
+ [
35
+ 2,
36
+ 50359
37
+ ],
38
+ [
39
+ 3,
40
+ 50363
41
+ ]
42
+ ],
43
+ "init_std": 0.02,
44
+ "is_encoder_decoder": true,
45
+ "mask_feature_length": 10,
46
+ "mask_feature_min_masks": 0,
47
+ "mask_feature_prob": 0.0,
48
+ "mask_time_length": 10,
49
+ "mask_time_min_masks": 2,
50
+ "mask_time_prob": 0.05,
51
+ "max_length": 448,
52
+ "max_source_positions": 1500,
53
+ "max_target_positions": 448,
54
+ "median_filter_width": 7,
55
+ "model_type": "whisper",
56
+ "neuron": {
57
+ "auto_cast": "matmul",
58
+ "auto_cast_type": "bf16",
59
+ "compiler_type": "neuronx-cc",
60
+ "compiler_version": "2.16.372.0+4a9b2326",
61
+ "disable_fallback": false,
62
+ "disable_fast_relayout": false,
63
+ "dynamic_batch_size": false,
64
+ "inline_weights_to_neff": true,
65
+ "input_names": [
66
+ "input_features",
67
+ "decoder_input_ids"
68
+ ],
69
+ "model_type": "whisper-encoder",
70
+ "optlevel": "2",
71
+ "output_attentions": false,
72
+ "output_hidden_states": false,
73
+ "output_names": [
74
+ "lm_logits",
75
+ "encoder_last_hidden_state"
76
+ ],
77
+ "static_batch_size": 1,
78
+ "static_nb_max_frames": 3000,
79
+ "static_sequence_length": 128,
80
+ "tensor_parallel_size": 1
81
+ },
82
+ "num_hidden_layers": 4,
83
+ "num_mel_bins": 80,
84
+ "pad_token_id": 50257,
85
+ "scale_embedding": false,
86
+ "stride": [
87
+ 1,
88
+ 2
89
+ ],
90
+ "suppress_tokens": [
91
+ 1,
92
+ 2,
93
+ 7,
94
+ 8,
95
+ 9,
96
+ 10,
97
+ 14,
98
+ 25,
99
+ 26,
100
+ 27,
101
+ 28,
102
+ 29,
103
+ 31,
104
+ 58,
105
+ 59,
106
+ 60,
107
+ 61,
108
+ 62,
109
+ 63,
110
+ 90,
111
+ 91,
112
+ 92,
113
+ 93,
114
+ 359,
115
+ 503,
116
+ 522,
117
+ 542,
118
+ 873,
119
+ 893,
120
+ 902,
121
+ 918,
122
+ 922,
123
+ 931,
124
+ 1350,
125
+ 1853,
126
+ 1982,
127
+ 2460,
128
+ 2627,
129
+ 3246,
130
+ 3253,
131
+ 3268,
132
+ 3536,
133
+ 3846,
134
+ 3961,
135
+ 4183,
136
+ 4667,
137
+ 6585,
138
+ 6647,
139
+ 7273,
140
+ 9061,
141
+ 9383,
142
+ 10428,
143
+ 10929,
144
+ 11938,
145
+ 12033,
146
+ 12331,
147
+ 12562,
148
+ 13793,
149
+ 14157,
150
+ 14635,
151
+ 15265,
152
+ 15618,
153
+ 16553,
154
+ 16604,
155
+ 18362,
156
+ 18956,
157
+ 20075,
158
+ 21675,
159
+ 22520,
160
+ 26130,
161
+ 26161,
162
+ 26435,
163
+ 28279,
164
+ 29464,
165
+ 31650,
166
+ 32302,
167
+ 32470,
168
+ 36865,
169
+ 42863,
170
+ 47425,
171
+ 49870,
172
+ 50254,
173
+ 50258,
174
+ 50358,
175
+ 50359,
176
+ 50360,
177
+ 50361,
178
+ 50362
179
+ ],
180
+ "task": "automatic-speech-recognition",
181
+ "torch_dtype": "float32",
182
+ "torchscript": true,
183
+ "transformers_version": "4.49.0",
184
+ "use_cache": true,
185
+ "use_weighted_layer_sum": false,
186
+ "vocab_size": 51865
187
+ }