Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2024-11-27 17:00:54 +01:00
Code Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
RVC-Boss 2023-08-31 17:51:55 +08:00 committed by GitHub
parent ca2308fd2f
commit 259ea663d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
infer/modules/vc/utils.py
View File

@ -2,22 +2,32 @@ import os
from fairseq import checkpoint_utils from fairseq import checkpoint_utils
### don't modify the code before you test it
# def get_index_path_from_model(sid):
# return next(
# (
# f
# for f in [
# os.path.join(root, name)
# for root, dirs, files in os.walk(os.getenv("index_root"), topdown=False)
# for name in files
# if name.endswith(".index") and "trained" not in name
# ]
# if sid.split(".")[0] in f
# ),
# "",
# )
def get_index_path_from_model(sid): def get_index_path_from_model(sid):
return next( sel_index_path = ""
( name = os.path.join("logs", sid.split(".")[0], "")
f # print(name)
for f in [ for f in index_paths:
os.path.join(root, name) if name in f:
for root, dirs, files in os.walk(os.getenv("index_root"), topdown=False) # print("selected index path:", f)
for name in files sel_index_path = f
if name.endswith(".index") and "trained" not in name break
] return sel_index_path
if sid.split(".")[0] in f
),
"",
)
def load_hubert(config): def load_hubert(config):
models, _, _ = checkpoint_utils.load_model_ensemble_and_task( models, _, _ = checkpoint_utils.load_model_ensemble_and_task(