From b86b2938495ff72d4fd4dd6ad30eb073231153cb Mon Sep 17 00:00:00 2001
From: AIFSH <1509359472@qq.com>
Date: Thu, 24 Oct 2024 15:16:06 +0800
Subject: [PATCH] change path

---
 README.md   | 2 +-
 __init__.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 4c7c77e..27244b8 100644
--- a/README.md
+++ b/README.md
@@ -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
\ No newline at end of file
+weights will be downloaded from github automatically
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index 80ee1dd..64afc68 100644
--- a/__init__.py
+++ b/__init__.py
@@ -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/"