Spaces:
Sleeping
Sleeping
John Langley
commited on
Commit
·
a121558
1
Parent(s):
d405851
change to GPU
Browse files
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 = "
|
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="
|
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
|