lorneluo commited on
Commit
623f2ef
1 Parent(s): 2796eb3

clean UVR import

Browse files
Files changed (1) hide show
  1. args.py +17 -1
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 = {}