Spaces:
Running
Running
File size: 324 Bytes
88b57c0 5eccb9a 88b57c0 |
1 2 3 4 5 6 7 8 9 10 |
# Third-party
import torch
# Global Variables
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
ONNX_MODEL_PATH = "./pretrained_models/MDX_net/Kim_Vocal.onnx"
INPUT_FOLDER = "./datasets/input"
OUTPUT_FOLDER = "./datasets/output"
|