diff options
| author | yum <yum.food.vr@gmail.com> | 2023-12-13 13:54:55 -0800 |
|---|---|---|
| committer | yum <yum.food.vr@gmail.com> | 2023-12-13 13:54:57 -0800 |
| commit | 921b92a69f36502dc5eefd14ba3487c1bb49bb9d (patch) | |
| tree | 06c2adf532d34440f3fa85fe62de519a357f8942 /Scripts/requirements.txt | |
| parent | 859caec3d5c1b6aa9eee98571af3324b6ed1bd21 (diff) | |
Begin experimenting with flash-attention
Seems much faster than faster-whisper.
There are two issues:
* Requires NVIDIA 3000 series or higher.
* Incompatible with faster-whisper dependencies.
So it seems like we'll either need to toggle between two sets of
dependencies at runtime or have two environments.
Diffstat (limited to 'Scripts/requirements.txt')
| -rw-r--r-- | Scripts/requirements.txt | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/Scripts/requirements.txt b/Scripts/requirements.txt index 9224ba8..58db539 100644 --- a/Scripts/requirements.txt +++ b/Scripts/requirements.txt @@ -1,17 +1,20 @@ -ctranslate2 editdistance -faster-whisper@https://github.com/guillaumekln/faster-whisper/archive/78d57d73c5b4a76b32d1d5a415e4e7aea760295c.tar.gz +flash-attn==2.3.6 future==0.18.2 huggingface_hub==0.16.4 keyboard langcodes language-data openvr +onnxruntime +packaging pillow pyaudio pydub python-osc pyyaml sentence_splitter -transformers>=4.21.0 +transformers==4.35.2 +--extra-index-url https://download.pytorch.org/whl/cu121 +torch |