diff --git a/modules/whisper/insanely_fast_whisper_inference.py b/modules/whisper/insanely_fast_whisper_inference.py index c95a26f2..21eb930d 100644 --- 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 index 9825cc8a..b0360160 100644 --- 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;