Hugging Face's logo

pravin96
/
distil_whisper_en

Automatic Speech Recognition
Transformers
TensorBoard
Safetensors
whisper
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
pravin96 commited on
Commit
4d7a6b0
•
1 Parent(s): ccbdd15

Model save

Browse files
Files changed (2) hide show
  1. README.md +6 -4
  2. generation_config.json +78 -0
README.md CHANGED
@@ -1,9 +1,11 @@
1
  ---
2
  library_name: transformers
3
  license: mit
4
- base_model: distil-whisper/distil-medium.en
5
  tags:
6
  - generated_from_trainer
 
 
7
  model-index:
8
  - name: distil_whisper_en
9
  results: []
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # distil_whisper_en
16
 
17
- This model is a fine-tuned version of [distil-whisper/distil-medium.en](https://huggingface.co/distil-whisper/distil-medium.en) on the None dataset.
18
 
19
  ## Model description
20
 
@@ -41,7 +43,7 @@ The following hyperparameters were used during training:
41
  - total_train_batch_size: 4
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
- - lr_scheduler_warmup_steps: 800
45
  - training_steps: 250
46
  - mixed_precision_training: Native AMP
47
 
@@ -52,6 +54,6 @@ The following hyperparameters were used during training:
52
  ### Framework versions
53
 
54
  - Transformers 4.44.2
55
- - Pytorch 2.4.1+cu121
56
  - Datasets 3.0.2
57
  - Tokenizers 0.19.1
 
1
  ---
2
  library_name: transformers
3
  license: mit
4
+ base_model: distil-whisper/distil-small.en
5
  tags:
6
  - generated_from_trainer
7
+ datasets:
8
+ - generator
9
  model-index:
10
  - name: distil_whisper_en
11
  results: []
 
16
 
17
  # distil_whisper_en
18
 
19
+ This model is a fine-tuned version of [distil-whisper/distil-small.en](https://huggingface.co/distil-whisper/distil-small.en) on the generator dataset.
20
 
21
  ## Model description
22
 
 
43
  - total_train_batch_size: 4
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - lr_scheduler_warmup_steps: 500
47
  - training_steps: 250
48
  - mixed_precision_training: Native AMP
49
 
 
54
  ### Framework versions
55
 
56
  - Transformers 4.44.2
57
+ - Pytorch 2.5.0+cu121
58
  - Datasets 3.0.2
59
  - Tokenizers 0.19.1
generation_config.json CHANGED
@@ -1,4 +1,82 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "begin_suppress_tokens": [
3
  220,
4
  50256
 
1
  {
2
+ "alignment_heads": [
3
+ [
4
+ 6,
5
+ 6
6
+ ],
7
+ [
8
+ 7,
9
+ 0
10
+ ],
11
+ [
12
+ 7,
13
+ 3
14
+ ],
15
+ [
16
+ 7,
17
+ 8
18
+ ],
19
+ [
20
+ 8,
21
+ 2
22
+ ],
23
+ [
24
+ 8,
25
+ 5
26
+ ],
27
+ [
28
+ 8,
29
+ 7
30
+ ],
31
+ [
32
+ 9,
33
+ 0
34
+ ],
35
+ [
36
+ 9,
37
+ 4
38
+ ],
39
+ [
40
+ 9,
41
+ 8
42
+ ],
43
+ [
44
+ 9,
45
+ 10
46
+ ],
47
+ [
48
+ 10,
49
+ 0
50
+ ],
51
+ [
52
+ 10,
53
+ 1
54
+ ],
55
+ [
56
+ 10,
57
+ 2
58
+ ],
59
+ [
60
+ 10,
61
+ 3
62
+ ],
63
+ [
64
+ 10,
65
+ 6
66
+ ],
67
+ [
68
+ 10,
69
+ 11
70
+ ],
71
+ [
72
+ 11,
73
+ 2
74
+ ],
75
+ [
76
+ 11,
77
+ 4
78
+ ]
79
+ ],
80
  "begin_suppress_tokens": [
81
  220,
82
  50256