John Langley commited on
Commit
a121558
·
1 Parent(s): d405851

change to GPU

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from utils import get_sentence, tts_interface
14
  # "cuda" - NVIDIA GPU
15
  # "cpu" - Plain CPU
16
  # "mps" - Apple silicon
17
- device = "cpu"
18
 
19
  # Load Mistral LLM
20
  print("Loading Mistral LLM")
@@ -24,7 +24,7 @@ mistral_llm = Llama(model_path=mistral_model_path,n_gpu_layers=35,max_new_tokens
24
 
25
  # Load Whisper ASR model
26
  print("Loading Whisper ASR")
27
- whisper_model = WhisperModel("large-v3", device="cpu", compute_type="float32")
28
 
29
 
30
  # Get all available voices from edge_tts
 
14
  # "cuda" - NVIDIA GPU
15
  # "cpu" - Plain CPU
16
  # "mps" - Apple silicon
17
+ device = "cuda"
18
 
19
  # Load Mistral LLM
20
  print("Loading Mistral LLM")
 
24
 
25
  # Load Whisper ASR model
26
  print("Loading Whisper ASR")
27
+ whisper_model = WhisperModel("large-v3", device="cuda", compute_type="float16")
28
 
29
 
30
  # Get all available voices from edge_tts