Skip to content

Commit

Permalink
Merge pull request #57 from pavelzbornik/dev
Browse files Browse the repository at this point in the history
Update dependencies and improve README for model support
  • Loading branch information
pavelzbornik authored Jan 6, 2025
2 parents 51dbb40 + 35201cf commit ea87c5b
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 9 deletions.
8 changes: 1 addition & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -75,9 +75,7 @@ WhisperX supports these model sizes:
- `base`, `base.en`
- `small`, `small.en`
- `medium`, `medium.en`
- `large`, `large-v1`, `large-v2`, `large-v3`

Note: `large-v3-turbo` is not yet supported by WhisperX.
- `large`, `large-v1`, `large-v2`, `large-v3`, `large-v3-turbo`

Set default model in `.env` using `WHISPER_MODEL=` (default: tiny)

Expand Down Expand Up @@ -185,10 +183,6 @@ The models used by whisperX are stored in `root/.cache`, if you want to avoid do
- Only `ctranslate2==4.4.0` is supported due to CUDA compatibility issues with CTranslate2, as newer versions have different CUDA requirements <https://github.com/SYSTRAN/faster-whisper/issues/1086>.
2. **faster-whisper Compatibility**
- Only `faster-whisper==1.0.0` is supported due to compatibility issues with WhisperX.
## Troubleshooting
### Common Issues
Expand Down
2 changes: 1 addition & 1 deletion dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
FROM nvidia/cuda:12.6.3-base-ubuntu22.04

ENV PYTHON_VERSION=3.11

Expand Down
2 changes: 1 addition & 1 deletion requirements/prod.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ gunicorn==23.0.0
httpx==0.28.1
numba==0.60.0

pytest==8.3.3
pytest==8.3.4
python-dotenv==1.0.1
python-multipart==0.0.20
tqdm==4.67.1
Expand Down

0 comments on commit ea87c5b

Please sign in to comment.