fix dependencies for faster-whisper 1.0.3 and ctranslate 4.4.0

AaltoRSE · Nov 1, 2024 · 920c2dd · 920c2dd
1 parent 055779d
commit 920c2dd
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/env.yml b/env.yml
@@ -9,12 +9,16 @@ dependencies:
   - libsndfile
   - python=3.10
   - pydub
+  - pytorch-lightning=2.2.5
   - nvidia::cuda-libraries-dev
-  - pytorch::pytorch
+  - pytorch::pytorch=2.3.1=py3.10_cuda12.1_cudnn8.9.2_0
   - pytorch::torchvision
   - pytorch::torchaudio
-  - pytorch::pytorch-cuda
+  - pytorch::pytorch-cuda=12.1
   - pip
   - pip:
     - pyannote.audio
-    - whisperx @ git+https://github.com/m-bain/whisperx.git
+    # Requires ctranslate 4.4.0 and faster-whisper 1.0.3. Link to the issue: https://github.com/m-bain/whisperX/issues/901
+    # "https://github.com/m-bain/whisperX/commit/caa7121064c1bb406be30c50891a9b8217252592" or
+    # "https://github.com/m-bain/whisperX/commit/5080b7188c3666b5d8648346c0c12599a58bd695"
+    - whisperx @ git+https://github.com/federicotorrielli/BetterWhisperX