Update infer/modules/vc/pipeline.py
Browse files
infer/modules/vc/pipeline.py
CHANGED
|
@@ -144,12 +144,13 @@ class Pipeline(object):
|
|
| 144 |
from infer.lib.rmvpe import RMVPE
|
| 145 |
|
| 146 |
logger.info(
|
| 147 |
-
"Loading rmvpe model,%s" % "%s/rmvpe.pt" % os.environ["rmvpe_root"]
|
| 148 |
)
|
| 149 |
self.model_rmvpe = RMVPE(
|
| 150 |
-
"%s/rmvpe.pt" % os.environ["rmvpe_root"],
|
| 151 |
is_half=self.is_half,
|
| 152 |
device=self.device,
|
|
|
|
| 153 |
)
|
| 154 |
f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
|
| 155 |
|
|
|
|
| 144 |
from infer.lib.rmvpe import RMVPE
|
| 145 |
|
| 146 |
logger.info(
|
| 147 |
+
"Loading rmvpe model,%s" % "%s/hpa-rmvpe.pt" % os.environ["rmvpe_root"]
|
| 148 |
)
|
| 149 |
self.model_rmvpe = RMVPE(
|
| 150 |
+
"%s/hpa-rmvpe.pt" % os.environ["rmvpe_root"],
|
| 151 |
is_half=self.is_half,
|
| 152 |
device=self.device,
|
| 153 |
+
hpa=True,
|
| 154 |
)
|
| 155 |
f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
|
| 156 |
|