nodes/SayanoAI-Comfy-RVC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed local model selection

Browse Source
This commit is contained in:
SayanoAI
2024-05-27 13:21:08 -04:00
parent 088bef921f
commit 8e3053e482
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_nodes/settings/downloader.py
+1 -1
View File
@@ -10,7 +10,7 @@ from ...lib import BASE_CACHE_DIR, BASE_MODELS_DIR
RVC_DOWNLOAD_LINK = 'https://huggingface.co/datasets/SayanoAI/RVC-Studio/resolve/main/'
MDX_MODELS = ["MDXNET/UVR-MDX-NET-vocal_FT.onnx"]
KARAFAN_MODELS = ["karafan/MDX23C-8KFFT-InstVoc_HQ.ckpt"]
VR_MODELS = ["UVR/UVR-DeEcho-DeReverb.pth","UVR/HP5-vocals+instrumentals.pth"]
VR_MODELS = ["UVR/UVR-DeEcho-DeReverb.pth","UVR/HP5-vocals+instrumentals.pth","UVR/5_HP-Karaoke-UVR.pth"]
RVC_MODELS = [
"RVC/Claire.pth",
"RVC/Sayano.pth",
custom_nodes/uvr.py
+2 -1
View File
@@ -20,7 +20,8 @@ class UVR5Node:
@classmethod
def INPUT_TYPES(cls):
model_list = MDX_MODELS + VR_MODELS + KARAFAN_MODELS + get_filenames(root=BASE_MODELS_DIR,format_func=os.path.basename)
model_list = MDX_MODELS + VR_MODELS + KARAFAN_MODELS + get_filenames(root=BASE_MODELS_DIR,format_func=lambda x: f"{os.path.basename(os.path.dirname(x))}/{os.path.basename(x)}",name_filters=["UVR","MDXNET","karafan"])
model_list = list(set(model_list)) # dedupe
return {
"required": {