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=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,