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
|