fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-05 01:10:22 +08:00
Code Activity

optimize(vc): apply model-style import

Browse Source
This commit is contained in:
源文雨
2024-06-02 00:57:03 +09:00
parent b0c33d1b75
commit 2991e75fc0
7 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
infer/modules/vc/__init__.py
View File

@@ -0,0 +1,3 @@
from .pipeline import Pipeline
from .modules import VC
from .utils import get_index_path_from_model, load_hubert

6
infer/modules/vc/modules.py
View File

@@ -1,5 +1,6 @@
import traceback
import logging
import os
logger = logging.getLogger(__name__)
@@ -15,8 +16,9 @@ from infer.lib.infer_pack.models import (
SynthesizerTrnMs768NSFsid,
SynthesizerTrnMs768NSFsid_nono,
)
from infer.modules.vc.pipeline import Pipeline
from infer.modules.vc.utils import *
from .pipeline import Pipeline
from .utils import get_index_path_from_model, load_hubert
class VC:

2
infer/modules/vc/pipeline.py
View File

@@ -144,7 +144,7 @@ class Pipeline(object):
from infer.lib.rmvpe import RMVPE
logger.info(
"Loading rmvpe model,%s" % "%s/rmvpe.pt" % os.environ["rmvpe_root"]
"Loading rmvpe model %s" % "%s/rmvpe.pt" % os.environ["rmvpe_root"]
)
self.model_rmvpe = RMVPE(
"%s/rmvpe.pt" % os.environ["rmvpe_root"],