Merge pull request #318 from jhj0517/fix/cudnn-problem

Fix/cudnn problem
jhj0517 · Oct 4, 2024 · 20f48aa · 20f48aa
2 parents 8faa542 + c8e54db
commit 20f48aa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/modules/whisper/insanely_fast_whisper_inference.py b/modules/whisper/insanely_fast_whisper_inference.py
@@ -78,7 +78,8 @@ def transcribe(self,
             kwargs = {
                 "no_speech_threshold": params.no_speech_threshold,
                 "temperature": params.temperature,
-                "compression_ratio_threshold": params.compression_ratio_threshold
+                "compression_ratio_threshold": params.compression_ratio_threshold,
+                "logprob_threshold": params.log_prob_threshold,
             }
 
             if self.current_model_size.endswith(".en"):

diff --git a/requirements.txt b/requirements.txt
@@ -2,14 +2,15 @@
 # If you're using it, update url to your CUDA version (CUDA 12.1 is minimum requirement):
 # For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
 # For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
---extra-index-url https://download.pytorch.org/whl/cu124
+--extra-index-url https://download.pytorch.org/whl/cu121
 
 
-torch
+torch==2.3.1
+torchaudio==2.3.1
 git+https://github.com/jhj0517/jhj0517-whisper.git
 faster-whisper==1.0.3
-transformers==4.42.3
-gradio==4.43.0
+transformers
+gradio
 pytubefix
 ruamel.yaml==0.18.6
 pyannote.audio==3.3.1