Views
No views yet
safetensors format for use with MLX on Apple Silicon.safetensors via the included convert_deepfilternet.py script| File | Description |
|---|---|
config.json | Model architecture configuration |
model.safetensors | Pre-converted weights (8.3 MB, float32) |
convert_deepfilternet.py | Conversion script (PyTorch → MLX safetensors) |
| Parameter | Value |
|---|---|
| Sample rate | 48 kHz |
| FFT size | 960 |
| Hop size | 480 |
| ERB bands | 32 |
| DF bins | 96 |
| DF order | 5 |
| Parameters | ~2M |
1import MLXAudioSTS
2
3let model = try await DeepFilterNetModel.fromPretrained("iky1e/DeepFilterNet3-MLX")
4let enhanced = try model.enhance(audioArray)1from mlx_audio.sts.models.deepfilternet import DeepFilterNetModel
2
3model = DeepFilterNetModel.from_pretrained(version=3, model_dir="path/to/local/dir")
4enhanced = model.enhance("noisy.wav")1python convert_deepfilternet.py \
2 --input /path/to/DeepFilterNet3 \
3 --output ./DeepFilterNet3-MLX \
4 --name DeepFilterNet3config.ini and a checkpoints/ folder from the original repo.1@inproceedings{schroeter2023deepfilternet3,
2 title={DeepFilterNet: Perceptually Motivated Real-Time Speech Enhancement},
3 author={Schr{\"o}ter, Hendrik and Rosenkranz, Tobias and Escalante-B., Alberto N. and Maier, Andreas},
4 booktitle={INTERSPEECH},
5 year={2023}
6}