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 09:10:25 +08:00
Code Activity

fix: outside_index_root detection

Browse Source
This commit is contained in:
源文雨
2024-06-02 00:19:19 +09:00
parent dc09d0e13e
commit b0c33d1b75
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,7 +9,8 @@ def get_index_path_from_model(sid):
f
for f in [
os.path.join(root, name)
for root, _, files in os.walk(os.getenv("index_root"), topdown=False)
for path in [os.getenv("outside_index_root"), os.getenv("index_root")]
for root, _, files in os.walk(path, topdown=False)
for name in files
if name.endswith(".index") and "trained" not in name
]
@@ -19,14 +20,14 @@ def get_index_path_from_model(sid):
)
def load_hubert(config):
def load_hubert(device, is_half):
models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
["assets/hubert/hubert_base.pt"],
suffix="",
)
hubert_model = models[0]
hubert_model = hubert_model.to(config.device)
if config.is_half:
hubert_model = hubert_model.to(device)
if is_half:
hubert_model = hubert_model.half()
else:
hubert_model = hubert_model.float()