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-06 09:40:24 +08:00
Code Activity

feat(infer): add model hash identification

Browse Source 
and optimize infer-web ui
This commit is contained in:
源文雨
2024-06-02 22:47:52 +09:00
parent 7e48279c6c
commit b9ad0258ae
13 changed files with 327 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -16,7 +16,7 @@ from infer.lib.infer_pack.models import (
SynthesizerTrnMs768NSFsid,
SynthesizerTrnMs768NSFsid_nono,
)
from .info import show_model_info
from .pipeline import Pipeline
from .utils import get_index_path_from_model, load_hubert
@@ -136,6 +136,7 @@ class VC:
to_return_protect1,
index,
index,
show_model_info(self.cpt)
)
if to_return_protect
else {"visible": True, "maximum": n_spk, "__type__": "update"}
@@ -158,6 +159,8 @@ class VC:
):
if input_audio_path is None:
return "You need to upload an audio", None
elif hasattr(input_audio_path, "name"):
input_audio_path = str(input_audio_path.name)
f0_up_key = int(f0_up_key)
try:
audio = load_audio(input_audio_path, 16000)
@@ -170,6 +173,7 @@ class VC:
self.hubert_model = load_hubert(self.config.device, self.config.is_half)
if file_index:
if hasattr(file_index, "name"): file_index = str(file_index.name)
file_index = (
file_index.strip(" ")
.strip('"')
@@ -207,12 +211,12 @@ class VC:
else:
tgt_sr = self.tgt_sr
index_info = (
"Index:\n%s." % file_index
"Index: %s." % file_index
if os.path.exists(file_index)
else "Index not used."
)
return (
"Success.\n%s\nTime:\nnpy: %.2fs, f0: %.2fs, infer: %.2fs."
"Success.\n%s\nTime: npy: %.2fs, f0: %.2fs, infer: %.2fs."
% (index_info, *times),
(tgt_sr, audio_opt),
)