diff --git a/custom_nodes/settings/downloader.py b/custom_nodes/settings/downloader.py index fab4a96..7d7d679 100644 --- a/custom_nodes/settings/downloader.py +++ b/custom_nodes/settings/downloader.py @@ -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", diff --git a/custom_nodes/uvr.py b/custom_nodes/uvr.py index 386ce47..fd36b3b 100644 --- a/custom_nodes/uvr.py +++ b/custom_nodes/uvr.py @@ -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": {