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

modules/whisper/insanely_fast_whisper_inference.py

@@ -78,7 +78,8 @@ class InsanelyFastWhisperInference(WhisperBase):
             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"):

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