cpu

hubert/inference.py

@@ -11,7 +11,6 @@ from hubert import hubert_model
 def load_model(path, device):
     model = hubert_model.hubert_soft(path)
     model.eval()
-    model.half()
     model.to(device)
     return model
 
@@ -19,7 +18,7 @@ def load_model(path, device):
 def pred_vec(model, wavPath, vecPath, device):
     feats = load_audio(wavPath)
     feats = torch.from_numpy(feats).to(device)
-    feats = feats[None, None, :].half()
+    feats = feats[None, None, :]
     with torch.no_grad():
         vec = model.units(feats).squeeze().data.cpu().float().numpy()
         # print(vec.shape)   # [length, dim=256] hop=320
@@ -38,8 +37,7 @@ if __name__ == "__main__":
     wavPath = args.wav
     vecPath = args.vec
 
-    assert torch.cuda.is_available()
-    device = "cuda"
+    device = "cpu"
     hubert = load_model(os.path.join(
         "hubert_pretrain", "hubert-soft-0d54a1f4.pt"), device)
     pred_vec(hubert, wavPath, vecPath, device)