Jarod Castillo commited on
Commit
d24a3d9
1 Parent(s): 4d09908

removed pip from requirements, fixed pretrained model pathing

Browse files
Files changed (3) hide show
  1. requirements.txt +0 -1
  2. src/constants.py +1 -1
  3. src/infer.py +1 -1
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- pip~=23.2.1
2
  torch~=2.0.1
3
  onnxruntime~=1.15.1
4
  librosa~=0.10.0.post2
 
 
1
  torch~=2.0.1
2
  onnxruntime~=1.15.1
3
  librosa~=0.10.0.post2
src/constants.py CHANGED
@@ -4,6 +4,6 @@ import torch
4
  # Global Variables
5
  COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
6
  EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
7
- ONNX_MODEL_PATH = "./models/MDX_Net_Models/Kim_Vocal.onnx"
8
  INPUT_FOLDER = "./datasets/input"
9
  OUTPUT_FOLDER = "./datasets/output"
 
4
  # Global Variables
5
  COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
6
  EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
7
+ ONNX_MODEL_PATH = "./pretrained_models/MDX_Net_Models/Kim_Vocal.onnx"
8
  INPUT_FOLDER = "./datasets/input"
9
  OUTPUT_FOLDER = "./datasets/output"
src/infer.py CHANGED
@@ -10,7 +10,7 @@ import onnxruntime as ort
10
  from models.MDX_net.mdx_net import Conv_TDF_net_trimm
11
  from loader import Loader
12
 
13
- vocal_path = r"./datasets/output/vocals.wav"
14
 
15
  # Global Variables
16
  COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
10
  from models.MDX_net.mdx_net import Conv_TDF_net_trimm
11
  from loader import Loader
12
 
13
+ vocal_path = "./datasets/output/vocals.wav"
14
 
15
  # Global Variables
16
  COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"