File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -136,15 +136,16 @@ strong performance in this language.
136
136
### Prepare Environment
137
137
138
138
We'll employ several popular Python packages to fine-tune the Whisper model.
139
- We'll use ` datasets ` to download and prepare our training data and
140
- ` transformers ` to load and train our Whisper model. We'll also require
141
- the ` soundfile ` package to pre-process audio files, ` evaluate ` and ` jiwer ` to
142
- assess the performance of our model. Finally, we'll
143
- use ` gradio ` to build a flashy demo of our fine-tuned model.
139
+ We'll use ` datasets ` to download and prepare our training data, alongside
140
+ ` transformers ` and ` accelerate ` to load and train our Whisper model.
141
+ We'll also require the ` soundfile ` package to pre-process audio files,
142
+ ` evaluate ` and ` jiwer ` to assess the performance of our model, and
143
+ ` tensorboard ` to log our metrics. Finally, we'll use ` gradio ` to build a
144
+ flashy demo of our fine-tuned model.
144
145
145
146
``` bash
146
147
! pip install --upgrade pip
147
- ! pip install --upgrade datasets transformers soundfile librosa evaluate jiwer gradio
148
+ ! pip install --upgrade datasets transformers accelerate soundfile librosa evaluate jiwer tensorboard gradio
148
149
```
149
150
150
151
We strongly advise you to upload model checkpoints directly the [ Hugging Face Hub] ( https://huggingface.co/ )
You can’t perform that action at this time.
0 commit comments