| from pathlib import Path |
|
|
|
|
| APP_ROOT = Path(__file__).resolve().parents[1] |
| RUNTIME_ROOT = Path("/tmp/sesa") |
| JOB_ROOT = RUNTIME_ROOT / "jobs" |
| MODEL_ROOT = RUNTIME_ROOT / "model-cache" |
| DOWNLOAD_ROOT = RUNTIME_ROOT / "downloads" |
| EXAMPLE_ROOT = RUNTIME_ROOT / "examples" |
| DATASET_SAMPLE_ROOT = EXAMPLE_ROOT / "dataset-samples" |
| MAX_DATASET_SAMPLE_FILE_BYTES = 500_000_000 |
| MAX_DATASET_SAMPLE_ARCHIVE_BYTES = 1_000_000_000 |
|
|
| MAX_FILES = 8 |
| MAX_MODELS = 6 |
| MAX_FILE_BYTES = 1_500_000_000 |
| MAX_WEIGHT_BYTES = 10_000_000_000 |
| MAX_CONFIG_BYTES = 5_000_000 |
| CPU_MAX_SECONDS = 120.0 |
| JOB_TTL_SECONDS = 6 * 60 * 60 |
|
|
| AUDIO_EXTENSIONS = { |
| ".wav", ".flac", ".mp3", ".ogg", ".opus", ".m4a", ".aiff", ".aif", ".ac3", ".wma" |
| } |
| VIDEO_EXTENSIONS = { |
| ".mp4", ".mov", ".mkv", ".webm", ".avi", ".mpeg", ".mpg", ".m4v" |
| } |
| INPUT_EXTENSIONS = AUDIO_EXTENSIONS | VIDEO_EXTENSIONS |
| OUTPUT_FORMATS = ["WAV", "FLAC", "MP3", "OGG", "OPUS", "M4A", "AIFF"] |
| ENSEMBLE_ALGORITHMS = [ |
| "avg_wave", "median_wave", "min_wave", "max_wave", |
| "avg_fft", "median_fft", "min_fft", "max_fft", |
| "uvr_max_spec", "uvr_min_spec", "ensemble_wav", |
| ] |
| STEM_CHOICES = ["All stems", "Vocals", "Instrumental", "Drums", "Bass", "Other", "Guitar", "Piano"] |
|
|
| SAFE_PACKAGE_MODELS = [ |
| ("MDX — Instrumental HQ 5", "UVR-MDX-NET-Inst_HQ_5.onnx"), |
| ("MDX — Instrumental HQ 3", "UVR-MDX-NET-Inst_HQ_3.onnx"), |
| ("MDX — Vocal FT", "UVR-MDX-NET-Voc_FT.onnx"), |
| ("MDX — Karaoke 2", "UVR_MDXNET_KARA_2.onnx"), |
| ("VR — De-Reverb", "UVR-De-Reverb-aufr33-jarredou.pth"), |
| ] |
|
|
| ALLOWED_INITIAL_HOSTS = { |
| "huggingface.co", |
| "www.huggingface.co", |
| "github.com", |
| "www.github.com", |
| "raw.githubusercontent.com", |
| } |
| ALLOWED_REDIRECT_HOSTS = ALLOWED_INITIAL_HOSTS | { |
| "cdn-lfs.hf.co", |
| "cas-bridge.xethub.hf.co", |
| "cas-server.xethub.hf.co", |
| "objects.githubusercontent.com", |
| "release-assets.githubusercontent.com", |
| } |
|
|
