Changing the fcpe checkpoint loading method"

Author: Vidalnt

rvc/lib/predictors/f0.py

@@ -2,7 +2,7 @@
 import torch
 
 from rvc.lib.predictors.RMVPE import RMVPE0Predictor
-from torchfcpe import spawn_bundled_infer_model
+from torchfcpe import spawn_infer_model_from_pt
 import torchcrepe
 from swift_f0 import SwiftF0
 import numpy as np
@@ -62,7 +62,7 @@ def __init__(self, device, sample_rate=16000, hop_size=160):
         self.device = device
         self.sample_rate = sample_rate
         self.hop_size = hop_size
-        self.model = spawn_bundled_infer_model(self.device)
+        self.model = spawn_infer_model_from_pt(os.path.join("rvc", "models", "predictors", "fcpe.pt"), self.device, bundled_model=True)
 
     def get_f0(self, x, p_len=None, filter_radius=0.006):
         if p_len is None: