sorendal commited on
Commit
bbee9f6
·
verified ·
1 Parent(s): bb61e60

Upload 5 files

Browse files
Files changed (4) hide show
  1. config.json +9 -9
  2. generation_config.json +21 -9
  3. model.safetensors +2 -2
  4. training_args.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "openai/whisper-base.en",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
@@ -13,17 +13,17 @@
13
  ],
14
  "bos_token_id": 50257,
15
  "classifier_proj_size": 256,
16
- "d_model": 512,
17
- "decoder_attention_heads": 8,
18
- "decoder_ffn_dim": 2048,
19
  "decoder_layerdrop": 0.0,
20
- "decoder_layers": 6,
21
  "decoder_start_token_id": 50257,
22
  "dropout": 0.0,
23
- "encoder_attention_heads": 8,
24
- "encoder_ffn_dim": 2048,
25
  "encoder_layerdrop": 0.0,
26
- "encoder_layers": 6,
27
  "eos_token_id": 50256,
28
  "forced_decoder_ids": [
29
  [
@@ -44,7 +44,7 @@
44
  "max_target_positions": 448,
45
  "median_filter_width": 7,
46
  "model_type": "whisper",
47
- "num_hidden_layers": 6,
48
  "num_mel_bins": 80,
49
  "pad_token_id": 50256,
50
  "scale_embedding": false,
 
1
  {
2
+ "_name_or_path": "openai/whisper-tiny.en",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
 
13
  ],
14
  "bos_token_id": 50257,
15
  "classifier_proj_size": 256,
16
+ "d_model": 384,
17
+ "decoder_attention_heads": 6,
18
+ "decoder_ffn_dim": 1536,
19
  "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 4,
21
  "decoder_start_token_id": 50257,
22
  "dropout": 0.0,
23
+ "encoder_attention_heads": 6,
24
+ "encoder_ffn_dim": 1536,
25
  "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 4,
27
  "eos_token_id": 50256,
28
  "forced_decoder_ids": [
29
  [
 
44
  "max_target_positions": 448,
45
  "median_filter_width": 7,
46
  "model_type": "whisper",
47
+ "num_hidden_layers": 4,
48
  "num_mel_bins": 80,
49
  "pad_token_id": 50256,
50
  "scale_embedding": false,
generation_config.json CHANGED
@@ -1,24 +1,36 @@
1
  {
2
  "alignment_heads": [
3
  [
4
- 3,
5
- 3
 
 
 
 
6
  ],
7
  [
8
- 4,
9
- 7
 
 
 
 
10
  ],
11
  [
12
- 5,
13
  1
14
  ],
15
  [
16
- 5,
17
- 5
 
 
 
 
18
  ],
19
  [
20
- 5,
21
- 7
22
  ]
23
  ],
24
  "begin_suppress_tokens": [
 
1
  {
2
  "alignment_heads": [
3
  [
4
+ 1,
5
+ 0
6
+ ],
7
+ [
8
+ 2,
9
+ 0
10
  ],
11
  [
12
+ 2,
13
+ 5
14
+ ],
15
+ [
16
+ 3,
17
+ 0
18
  ],
19
  [
20
+ 3,
21
  1
22
  ],
23
  [
24
+ 3,
25
+ 2
26
+ ],
27
+ [
28
+ 3,
29
+ 3
30
  ],
31
  [
32
+ 3,
33
+ 4
34
  ]
35
  ],
36
  "begin_suppress_tokens": [
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3b872ae9e51a2f42865dc087473340af8232ec6996952e6c3a1616854d325807
3
- size 290401888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8577ce21d5a05cc7a534570d9183f433f559cb3eb0ac59644a5df2b352277335
3
+ size 151060136
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a91687ebe726f62be112063606b59ced02c74621a585eb85ae736307a269b1e5
3
  size 5496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:547dd548294e64c5fb8255c205bd4e8a0f247352096fd5c661d29d23111fc5b8
3
  size 5496