Spaces:
Runtime error
Runtime error
load_model on cpu
Browse files
src/models/whisper/audio2feature.py
CHANGED
|
@@ -11,7 +11,7 @@ class Audio2Feature():
|
|
| 11 |
model_path="./models/whisper/tiny.pt",
|
| 12 |
device="cuda"):
|
| 13 |
self.whisper_model_type = whisper_model_type
|
| 14 |
-
self.model = load_model(model_path, device=
|
| 15 |
|
| 16 |
def get_sliced_feature(self,
|
| 17 |
feature_array,
|
|
|
|
| 11 |
model_path="./models/whisper/tiny.pt",
|
| 12 |
device="cuda"):
|
| 13 |
self.whisper_model_type = whisper_model_type
|
| 14 |
+
self.model = load_model(model_path, device='cpu') #
|
| 15 |
|
| 16 |
def get_sliced_feature(self,
|
| 17 |
feature_array,
|