rosyvs commited on
Commit
8a14c0d
1 Parent(s): 7bf3b60

swap for non-int8 version

Browse files
Files changed (3) hide show
  1. config.json +104 -20
  2. generation_config.json +1 -1
  3. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "openai/whisper-medium.en",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
@@ -25,13 +25,18 @@
25
  "encoder_layerdrop": 0.0,
26
  "encoder_layers": 24,
27
  "eos_token_id": 50256,
28
- "forced_decoder_ids": null,
 
 
 
 
 
29
  "init_std": 0.02,
30
  "is_encoder_decoder": true,
31
- "mask_feature_length": 30,
32
- "mask_feature_min_masks": 2,
33
- "mask_feature_prob": 0.05,
34
- "mask_time_length": 40,
35
  "mask_time_min_masks": 2,
36
  "mask_time_prob": 0.05,
37
  "max_length": 448,
@@ -42,22 +47,101 @@
42
  "num_hidden_layers": 24,
43
  "num_mel_bins": 80,
44
  "pad_token_id": 50256,
45
- "quantization_config": {
46
- "bnb_4bit_compute_dtype": "float32",
47
- "bnb_4bit_quant_type": "fp4",
48
- "bnb_4bit_use_double_quant": false,
49
- "llm_int8_enable_fp32_cpu_offload": false,
50
- "llm_int8_has_fp16_weight": false,
51
- "llm_int8_skip_modules": null,
52
- "llm_int8_threshold": 6.0,
53
- "load_in_4bit": false,
54
- "load_in_8bit": true,
55
- "quant_method": "bitsandbytes"
56
- },
57
  "scale_embedding": false,
58
- "suppress_tokens": [],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
  "torch_dtype": "float32",
60
- "transformers_version": "4.32.1",
61
  "use_cache": false,
62
  "use_weighted_layer_sum": false,
63
  "vocab_size": 51864
 
1
  {
2
+ "_name_or_path": "/shared/models/LEVI_LoFi_v2_MediumEN_Lora_Int8",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
 
25
  "encoder_layerdrop": 0.0,
26
  "encoder_layers": 24,
27
  "eos_token_id": 50256,
28
+ "forced_decoder_ids": [
29
+ [
30
+ 1,
31
+ 50362
32
+ ]
33
+ ],
34
  "init_std": 0.02,
35
  "is_encoder_decoder": true,
36
+ "mask_feature_length": 10,
37
+ "mask_feature_min_masks": 0,
38
+ "mask_feature_prob": 0.0,
39
+ "mask_time_length": 10,
40
  "mask_time_min_masks": 2,
41
  "mask_time_prob": 0.05,
42
  "max_length": 448,
 
47
  "num_hidden_layers": 24,
48
  "num_mel_bins": 80,
49
  "pad_token_id": 50256,
 
 
 
 
 
 
 
 
 
 
 
 
50
  "scale_embedding": false,
51
+ "suppress_tokens": [
52
+ 1,
53
+ 2,
54
+ 7,
55
+ 8,
56
+ 9,
57
+ 10,
58
+ 14,
59
+ 25,
60
+ 26,
61
+ 27,
62
+ 28,
63
+ 29,
64
+ 31,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 90,
72
+ 91,
73
+ 92,
74
+ 93,
75
+ 357,
76
+ 366,
77
+ 438,
78
+ 532,
79
+ 685,
80
+ 705,
81
+ 796,
82
+ 930,
83
+ 1058,
84
+ 1220,
85
+ 1267,
86
+ 1279,
87
+ 1303,
88
+ 1343,
89
+ 1377,
90
+ 1391,
91
+ 1635,
92
+ 1782,
93
+ 1875,
94
+ 2162,
95
+ 2361,
96
+ 2488,
97
+ 3467,
98
+ 4008,
99
+ 4211,
100
+ 4600,
101
+ 4808,
102
+ 5299,
103
+ 5855,
104
+ 6329,
105
+ 7203,
106
+ 9609,
107
+ 9959,
108
+ 10563,
109
+ 10786,
110
+ 11420,
111
+ 11709,
112
+ 11907,
113
+ 13163,
114
+ 13697,
115
+ 13700,
116
+ 14808,
117
+ 15306,
118
+ 16410,
119
+ 16791,
120
+ 17992,
121
+ 19203,
122
+ 19510,
123
+ 20724,
124
+ 22305,
125
+ 22935,
126
+ 27007,
127
+ 30109,
128
+ 30420,
129
+ 33409,
130
+ 34949,
131
+ 40283,
132
+ 40493,
133
+ 40549,
134
+ 47282,
135
+ 49146,
136
+ 50257,
137
+ 50357,
138
+ 50358,
139
+ 50359,
140
+ 50360,
141
+ 50361
142
+ ],
143
  "torch_dtype": "float32",
144
+ "transformers_version": "4.34.0",
145
  "use_cache": false,
146
  "use_weighted_layer_sum": false,
147
  "vocab_size": 51864
generation_config.json CHANGED
@@ -185,6 +185,6 @@
185
  50360,
186
  50361
187
  ],
188
- "transformers_version": "4.32.1",
189
  "use_cache": false
190
  }
 
185
  50360,
186
  50361
187
  ],
188
+ "transformers_version": "4.34.0",
189
  "use_cache": false
190
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8e553c346e8f7c65be71f73a9741a6f00d24327c9344bfb043e3c972296f02a1
3
- size 3055738070
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7f66ea872fb5330ebabaaacdd61889156ec6bc203cc6040bd27096cf084fa7e
3
+ size 3055737561