Spaces:
Running
on
Zero
Running
on
Zero
wenmengzhou
commited on
Commit
•
8e96caf
1
Parent(s):
c1f181e
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,
|