Downloads · 30 days
40
4% of all-time downloads
OpenUnmix-HQ-MLX
aufklarer/OpenUnmix-HQ-MLX
OpenUnmix-HQ-MLX is a audio-to-audio model from aufklarer. Use it for the audio-to-audio task on the model card, and read the license before you ship it in a product. It is set up for mlx. The card lists the license as mit.
MLX-converted Open-Unmix HQ for music source separation on Apple Silicon.
All-time downloads
995
Public
Repo size
142 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors142 MB · 100%
At a glance
- Task
- Audio-to-Audio
- Library
- mlx
- License
- mit
- Access
- Public
- Created
- Mar 24, 2026
- Updated
- Mar 25, 2026
- SHA
- 1e9976cf
Datasets
Model card
From the Hugging Face model README
Open-Unmix HQ (MLX)
MLX-converted Open-Unmix HQ for music source separation on Apple Silicon.
Separates stereo music into 4 stems: vocals, drums, bass, other. 4.3x real-time on M2 Max.
Model Details
- Architecture: BiLSTM + FC encoder/decoder with magnitude masking
- Parameters: 8.9M per stem (4 stems = 35.6M total)
- Hidden size: 512
- Input: Stereo 44.1kHz audio
- Output: 4 stereo WAV stems
- Format: safetensors (MLX-compatible)
- Size: ~34 MB per stem, ~136 MB total
Benchmark (MUSDB18-HQ, 50 tracks, M2 Max)
| Target | SDR (dB) |
|---|---|
| Vocals | 6.23 |
| Drums | 6.44 |
| Bass | 4.56 |
| Other | 3.41 |
RTF 0.23 (4.3x real-time).
Usage
Used by speech-swift:
audio separate song.wav
let separator = try await SourceSeparator.fromPretrained()
let stems = separator.separate(audio: stereoAudio, sampleRate: 44100)
// stems[.vocals], stems[.drums], stems[.bass], stems[.other]
Files
vocals.safetensors— Vocals model (34 MB)drums.safetensors— Drums model (34 MB)bass.safetensors— Bass model (34 MB)other.safetensors— Other/accompaniment model (34 MB)config.json— Model configuration
Reference
- Open-Unmix (GitHub)
- Stöter et al., "Open-Unmix — A Reference Implementation for Music Source Separation" (JOSS, 2019)
License
MIT (same as original Open-Unmix)
- Task
- Audio-to-Audio
- Library
- mlx
- License
- mit
- Languages
- en
- Created
- Mar 24, 2026
- Updated
- Mar 25, 2026