diff --git a/src/requirements.txt b/src/requirements.txt index 5f6b58e8..094d3276 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -14,7 +14,7 @@ unstructured==0.11.6 psutil==5.9.5 PySide6==6.6.1 PyAudio==0.2.14 -https://github.com/SYSTRAN/faster-whisper/archive/refs/tags/0.10.0.tar.gz +faster-whisper==0.10.1 termcolor==2.3.0 pypandoc==1.12 PyYAML==6.0.1 @@ -23,4 +23,4 @@ transformers==4.36.0 accelerate==0.25.0 optimum==1.15.0 einops==0.7.0 -scipy==1.11.4 +scipy==1.11.4 \ No newline at end of file diff --git a/src/setup.py b/src/setup.py index e68f5a78..5b9e43e0 100644 --- a/src/setup.py +++ b/src/setup.py @@ -54,7 +54,11 @@ def display_cuda_message(): return "12.1", tkinter_message_box("CUDA Check", "CUDA version 12.1 detected. CUDA 11.8 is required. Click OK to proceed with CPU-only installation of PyTorch or Cancel to exit installer.", type="yesno", yes_no=True) else: update_cuda = tkinter_message_box("CUDA Check", f"Incorrect version of CUDA installed (Version: {cuda_version}). Would you like to proceed with a CPU-only installation?", type="yesno", yes_no=True) - return None, update_cuda + if update_cuda: + return None, True + else: + print("Exiting installer.") + sys.exit(0) def manual_installation_confirmation(): if not tkinter_message_box("Confirmation", c.MESSAGE_GIT, type="yesno", yes_no=True):