Skip to content

Commit a136eaf

Browse files
[Whisper] Update requirements x2 (#1488)
1 parent 2f9fa7a commit a136eaf

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

fine-tune-whisper.md

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -136,15 +136,16 @@ strong performance in this language.
136136
### Prepare Environment
137137

138138
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.
144145

145146
```bash
146147
!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
148149
```
149150

150151
We strongly advise you to upload model checkpoints directly the [Hugging Face Hub](https://huggingface.co/)

0 commit comments

Comments
 (0)