Automatically select index file when model is selected (#894)

* automatically select index file when model is selected

* Search for full directory of index file

* Add trailing separator char to index search string

* disable debug log

* remove unused re

---------

Co-authored-by: 源文雨 <41315874+fumiama@users.noreply.github.com>
This commit is contained in:
Flynn Duniho 2023-07-29 20:13:03 -07:00 committed by GitHub
parent 0784b4e593
commit 144073a924
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 1 deletions

View File

@ -399,6 +399,16 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
torch.cuda.empty_cache()
yield "\n".join(infos)
def get_index_path_from_model(sid):
sel_index_path = ""
name = os.path.join("logs", sid.split(".")[0], "")
# print(name)
for f in index_paths:
if name in f:
# print("selected index path:", f)
sel_index_path = f
break
return sel_index_path
# 一个选项卡全局只能有一个音色
def get_vc(sid, to_return_protect0, to_return_protect1):
@ -434,6 +444,7 @@ def get_vc(sid, to_return_protect0, to_return_protect1):
return {"visible": False, "__type__": "update"}
person = "%s/%s" % (weight_root, sid)
print("loading %s" % person)
cpt = torch.load(person, map_location="cpu")
tgt_sr = cpt["config"][-1]
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] # n_spk
@ -479,6 +490,7 @@ def get_vc(sid, to_return_protect0, to_return_protect1):
{"visible": True, "maximum": n_spk, "__type__": "update"},
to_return_protect0,
to_return_protect1,
get_index_path_from_model(sid),
)
@ -1629,7 +1641,7 @@ with gr.Blocks(title="RVC WebUI") as app:
sid0.change(
fn=get_vc,
inputs=[sid0, protect0, protect1],
outputs=[spk_item, protect0, protect1],
outputs=[spk_item, protect0, protect1, file_index2],
)
with gr.TabItem(i18n("伴奏人声分离&去混响&去回声")):
with gr.Group():