Files

14 lines
394 B
Python

CATEGORY = "audio/pymss"
MSS_MAX_STEMS = 8
VR_MAX_STEMS = 2
MSS_PARAMS_TYPE = "PYMSS_MSS_PARAMS"
VR_PARAMS_TYPE = "PYMSS_VR_PARAMS"
MODEL_FOLDER_NAME = "pymss"
MODEL_DIR_ENV_VARS = ("COMFY_MSS_MODEL_DIR", "PYMSS_MODEL_DIR")
NOT_DOWNLOADED_PREFIX = "[Not downloaded] "
CUSTOM_MODEL_EXTENSIONS = (".ckpt", ".pth", ".pt", ".safetensors", ".safetensor", ".bin")
CUSTOM_MODEL_DIR_NAME = "custom"