Merge pull request #839 from AznamirWoW/pipeline_fix

Another performance improvement ~0.3sec/call
IAHispano · Oct 26, 2024 · 96f9e2d · 96f9e2d
2 parents 94dec75 + d5212fb
commit 96f9e2d
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/rvc/infer/pipeline.py b/rvc/infer/pipeline.py
@@ -207,7 +207,11 @@ def __init__(self, tgt_sr, config):
         ]
         self.autotune = Autotune(self.ref_freqs)
         self.note_dict = self.autotune.note_dict
-
+        self.model_rmvpe = RMVPE0Predictor(
+            os.path.join("rvc", "models", "predictors", "rmvpe.pt"),
+            is_half=self.is_half,
+            device=self.device,
+        )
     def get_f0_crepe(
         self,
         x,
@@ -291,11 +295,6 @@ def get_f0_hybrid(
                     x, f0_min, f0_max, p_len, int(hop_length)
                 )
             elif method == "rmvpe":
-                self.model_rmvpe = RMVPE0Predictor(
-                    os.path.join("rvc", "models", "predictors", "rmvpe.pt"),
-                    is_half=self.is_half,
-                    device=self.device,
-                )
                 f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
                 f0 = f0[1:]
             elif method == "fcpe":
@@ -356,11 +355,6 @@ def get_f0(
                 x, self.f0_min, self.f0_max, p_len, int(hop_length), "tiny"
             )
         elif f0_method == "rmvpe":
-            self.model_rmvpe = RMVPE0Predictor(
-                os.path.join("rvc", "models", "predictors", "rmvpe.pt"),
-                is_half=self.is_half,
-                device=self.device,
-            )
             f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
         elif f0_method == "fcpe":
             self.model_fcpe = FCPEF0Predictor(