Spaces:
Running
Running
Jarod Castillo
commited on
Commit
•
5eccb9a
1
Parent(s):
d24a3d9
fixed pretrained model pathing 2.0
Browse files- src/constants.py +1 -1
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 = "./pretrained_models/
|
| 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/Kim_Vocal.onnx"
|
| 8 |
INPUT_FOLDER = "./datasets/input"
|
| 9 |
OUTPUT_FOLDER = "./datasets/output"
|