gincioks commited on
Commit
d2ac737
1 Parent(s): 21e6f3d

Upload WhisperForConditionalGeneration

Browse files
Files changed (3) hide show
  1. config.json +42 -0
  2. generation_config.json +110 -0
  3. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./whisper-tiny-lt/checkpoint-1000",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "gelu",
5
+ "architectures": [
6
+ "WhisperForConditionalGeneration"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "begin_suppress_tokens": [
10
+ 220,
11
+ 50257
12
+ ],
13
+ "bos_token_id": 50257,
14
+ "d_model": 384,
15
+ "decoder_attention_heads": 6,
16
+ "decoder_ffn_dim": 1536,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 4,
19
+ "decoder_start_token_id": 50258,
20
+ "dropout": 0.0,
21
+ "encoder_attention_heads": 6,
22
+ "encoder_ffn_dim": 1536,
23
+ "encoder_layerdrop": 0.0,
24
+ "encoder_layers": 4,
25
+ "eos_token_id": 50257,
26
+ "forced_decoder_ids": null,
27
+ "init_std": 0.02,
28
+ "is_encoder_decoder": true,
29
+ "max_length": 448,
30
+ "max_source_positions": 1500,
31
+ "max_target_positions": 448,
32
+ "model_type": "whisper",
33
+ "num_hidden_layers": 4,
34
+ "num_mel_bins": 80,
35
+ "pad_token_id": 50257,
36
+ "scale_embedding": false,
37
+ "suppress_tokens": [],
38
+ "torch_dtype": "float32",
39
+ "transformers_version": "4.26.1",
40
+ "use_cache": true,
41
+ "vocab_size": 51865
42
+ }
generation_config.json ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "begin_suppress_tokens": [
3
+ 220,
4
+ 50257
5
+ ],
6
+ "bos_token_id": 50257,
7
+ "decoder_start_token_id": 50258,
8
+ "eos_token_id": 50257,
9
+ "forced_decoder_ids": [
10
+ [
11
+ 1,
12
+ null
13
+ ],
14
+ [
15
+ 2,
16
+ 50359
17
+ ]
18
+ ],
19
+ "max_length": 448,
20
+ "pad_token_id": 50256,
21
+ "suppress_tokens": [
22
+ 1,
23
+ 2,
24
+ 7,
25
+ 8,
26
+ 9,
27
+ 10,
28
+ 14,
29
+ 25,
30
+ 26,
31
+ 27,
32
+ 28,
33
+ 29,
34
+ 31,
35
+ 58,
36
+ 59,
37
+ 60,
38
+ 61,
39
+ 62,
40
+ 63,
41
+ 90,
42
+ 91,
43
+ 92,
44
+ 93,
45
+ 359,
46
+ 503,
47
+ 522,
48
+ 542,
49
+ 873,
50
+ 893,
51
+ 902,
52
+ 918,
53
+ 922,
54
+ 931,
55
+ 1350,
56
+ 1853,
57
+ 1982,
58
+ 2460,
59
+ 2627,
60
+ 3246,
61
+ 3253,
62
+ 3268,
63
+ 3536,
64
+ 3846,
65
+ 3961,
66
+ 4183,
67
+ 4667,
68
+ 6585,
69
+ 6647,
70
+ 7273,
71
+ 9061,
72
+ 9383,
73
+ 10428,
74
+ 10929,
75
+ 11938,
76
+ 12033,
77
+ 12331,
78
+ 12562,
79
+ 13793,
80
+ 14157,
81
+ 14635,
82
+ 15265,
83
+ 15618,
84
+ 16553,
85
+ 16604,
86
+ 18362,
87
+ 18956,
88
+ 20075,
89
+ 21675,
90
+ 22520,
91
+ 26130,
92
+ 26161,
93
+ 26435,
94
+ 28279,
95
+ 29464,
96
+ 31650,
97
+ 32302,
98
+ 32470,
99
+ 36865,
100
+ 42863,
101
+ 47425,
102
+ 49870,
103
+ 50254,
104
+ 50258,
105
+ 50360,
106
+ 50361,
107
+ 50362
108
+ ],
109
+ "transformers_version": "4.26.1"
110
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d04c1fbc95dd7b58dc14b2d19d7015b983f476c3190f7e1f30d5251e257c164
3
+ size 151096617