nodes/AIFSH-VocalSeparation-ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change path

Browse Source
This commit is contained in:
AIFSH
2024-10-24 15:16:06 +08:00
parent 649f4ca0a9
commit b86b293849
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -1
View File
@@ -8,4 +8,4 @@ git clone https://github.com/AIFSH/VocalSeparation-ComfyUI.git
cd VocalSeparation-ComfyUI
pip install -r requirements.txt
```
weights will be downloaded from github
weights will be downloaded from github automatically
__init__.py
+2 -1
View File
@@ -6,10 +6,11 @@ import torchaudio
import cuda_malloc
import numpy as np
import urllib
import folder_paths
from .mss.utils import get_model_from_config,demix_track,demix_track_demucs
now_dir = os.path.dirname(os.path.abspath(__file__))
models_dir = os.path.join(now_dir, "pretrained_models")
models_dir = os.path.join(folder_paths.models_dir,"AIFSH", "Music-Source-Separation-Training")
os.makedirs(models_dir,exist_ok=True)
base_url = "https://github.com/ZFTurbo/Music-Source-Separation-Training/releases/download/v1.0.0/"