clean UVR import
Browse files
args.py
CHANGED
@@ -5,9 +5,25 @@ from addict import Dict
|
|
5 |
|
6 |
from ml_collections import ConfigDict
|
7 |
|
8 |
-
from UVR import DENOISER_MODEL_PATH, DEVERBER_MODEL_PATH, MDX_C_CONFIG_PATH
|
9 |
from gui_data.constants import DEMUCS_2_SOURCE, DEMUCS_2_SOURCE_MAPPER, VR_ARCH_TYPE, MDX_ARCH_TYPE, DEMUCS_ARCH_TYPE
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
vr_cache_source_mapper = {}
|
12 |
mdx_cache_source_mapper = {}
|
13 |
demucs_cache_source_mapper = {}
|
|
|
5 |
|
6 |
from ml_collections import ConfigDict
|
7 |
|
|
|
8 |
from gui_data.constants import DEMUCS_2_SOURCE, DEMUCS_2_SOURCE_MAPPER, VR_ARCH_TYPE, MDX_ARCH_TYPE, DEMUCS_ARCH_TYPE
|
9 |
|
10 |
+
BASE_PATH = os.path.dirname(os.path.abspath(__file__))
|
11 |
+
MODELS_DIR = os.path.join(BASE_PATH, 'models')
|
12 |
+
VR_MODELS_DIR = os.path.join(MODELS_DIR, 'VR_Models')
|
13 |
+
MDX_MODELS_DIR = os.path.join(MODELS_DIR, 'MDX_Net_Models')
|
14 |
+
DEMUCS_MODELS_DIR = os.path.join(MODELS_DIR, 'Demucs_Models')
|
15 |
+
DEMUCS_NEWER_REPO_DIR = os.path.join(DEMUCS_MODELS_DIR, 'v3_v4_repo')
|
16 |
+
MDX_MIXER_PATH = os.path.join(BASE_PATH, 'lib_v5', 'mixer.ckpt')
|
17 |
+
DENOISER_MODEL_PATH = os.path.join(VR_MODELS_DIR, 'UVR-DeNoise-Lite.pth')
|
18 |
+
DEVERBER_MODEL_PATH = os.path.join(VR_MODELS_DIR, 'UVR-DeEcho-DeReverb.pth')
|
19 |
+
|
20 |
+
# Cache & Parameters
|
21 |
+
VR_HASH_DIR = os.path.join(VR_MODELS_DIR, 'model_data')
|
22 |
+
VR_HASH_JSON = os.path.join(VR_MODELS_DIR, 'model_data', 'model_data.json')
|
23 |
+
MDX_HASH_DIR = os.path.join(MDX_MODELS_DIR, 'model_data')
|
24 |
+
MDX_HASH_JSON = os.path.join(MDX_HASH_DIR, 'model_data.json')
|
25 |
+
MDX_C_CONFIG_PATH = os.path.join(MDX_HASH_DIR, 'mdx_c_configs')
|
26 |
+
|
27 |
vr_cache_source_mapper = {}
|
28 |
mdx_cache_source_mapper = {}
|
29 |
demucs_cache_source_mapper = {}
|