Spaces:
Runtime error
Runtime error
Update whisper_asr.py
Browse files- whisper_asr.py +5 -2
whisper_asr.py
CHANGED
@@ -7,8 +7,11 @@ import os
|
|
7 |
import time
|
8 |
|
9 |
class WhisperAutomaticSpeechRecognizer:
|
10 |
-
device = "cuda"
|
11 |
-
compute_type = "int8" # change to if more gpu memory available
|
|
|
|
|
|
|
12 |
batch_size = 4
|
13 |
model = whisperx.load_model(
|
14 |
"medium", device, language="en", compute_type=compute_type
|
|
|
7 |
import time
|
8 |
|
9 |
class WhisperAutomaticSpeechRecognizer:
|
10 |
+
# device = "cuda"
|
11 |
+
# compute_type = "int8" # change to if more gpu memory available
|
12 |
+
device = "cpu"
|
13 |
+
compute_type = "float32"
|
14 |
+
|
15 |
batch_size = 4
|
16 |
model = whisperx.load_model(
|
17 |
"medium", device, language="en", compute_type=compute_type
|