Skip to content

Commit

Permalink
Added download/convert option for distilled Whisper models
Browse files Browse the repository at this point in the history
  • Loading branch information
Aleks committed Mar 30, 2024
1 parent 033b008 commit 08e4f88
Show file tree
Hide file tree
Showing 6 changed files with 50,297 additions and 18 deletions.
2 changes: 1 addition & 1 deletion .env
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ DEBUG=True
# docker run command as a volume.
# e.g. WHISPER_MODEL="/app/models/custom"
# docker cmd: -v /path/to/custom/model:/app/models/custom
WHISPER_MODEL="distil-whisper-v2"
WHISPER_MODEL="distil-large-v2"
# You can specify one of two engines, "faster-whisper" or "tensorrt-llm".
WHISPER_ENGINE="tensorrt-llm"
# The compute_type parameter is used to control the precision of the model. You can choose between:
Expand Down
1 change: 1 addition & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ repos:
rev: "v4.3.0"
hooks:
- id: check-added-large-files
args: [--maxkb=2000]
- id: check-toml
- id: check-yaml

Expand Down
1 change: 1 addition & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ RUN apt update && apt install -y \
python3-pip \
libbz2-dev \
python3-dev \
liblzma-dev \
&& rm -rf /var/lib/apt/lists/*

RUN apt update && apt install -y \
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ dependencies = [
"soundfile==0.12.1",
"safetensors==0.4.2",
"janus==1.0.0",
"backports.lzma",
"backports.lzma==0.0.14",
]

[project.urls]
Expand Down
Loading

0 comments on commit 08e4f88

Please sign in to comment.