from mlx_audio.sts.models.mossformer2_se import MossFormer2SEModel

model = MossFormer2SEModel.from_pretrained("starkdmi/MossFormer2-SE-6bit")
enhanced = model.enhance("noisy.wav")

Precision Variants

MossFormer2-SE (fp32, 211MB)
MossFormer2-SE-fp16 (fp16, 106MB)
MossFormer2-SE-8bit (int8, 86MB)
MossFormer2-SE-6bit (int6, 75MB) ← this
MossFormer2-SE-4bit (int4, 64MB)

Performance

~30x real-time on Apple M4