Skip to content

Commit

Permalink
final cuda 11.8 release
Browse files Browse the repository at this point in the history
Correct faster-whisper dependency and setup to exit when a user chooses to.
  • Loading branch information
BBC-Esq authored Feb 22, 2024
1 parent a96b01c commit d2fa021
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
4 changes: 2 additions & 2 deletions src/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
6 changes: 5 additions & 1 deletion src/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit d2fa021

Please sign in to comment.