Update model/utils.py

model/utils.py

@@ -437,7 +437,7 @@ def load_asr_model(lang, ckpt_dir = ""):
     elif lang == "en":
         from faster_whisper import WhisperModel
         model_size = "large-v3" if ckpt_dir == "" else ckpt_dir
-        model = WhisperModel(model_size, device="cuda", compute_type="float16")
+        model = WhisperModel(model_size, device="cpu", compute_type="float16")
     return model
 
 
@@ -565,7 +565,7 @@ def load_checkpoint(model, ckpt_path, device, use_ema = True):
         from safetensors.torch import load_file
         checkpoint = load_file(ckpt_path, device=device)
     else:
-        checkpoint = torch.load(ckpt_path, weights_only=True, map_location=device)
+        checkpoint = torch.load(ckpt_path, weights_only=False, map_location=device)
 
     if use_ema == True:
         ema_model = EMA(model, include_online_model = False).to(device)