CoolFace
Modelpublic

aufklarer/WavLM-Base-Plus-MLX-fp16

sourceHugging Facemitupdated 3mo agoView on Hugging Face
0likes37downloads
Model Card

WavLM Base Plus MLX fp16

MLX-compatible safetensors export of microsoft/wavlm-base-plus. This bundle is used by speech-swift as the SSL feature-extractor companion for Indic-Mio raw-reference voice cloning.

WavLM runs on 16 kHz mono speech audio and produces hidden states that downstream speech models can use for speaker, content, and paralinguistic representations. For Indic-Mio, speech-swift averages hidden layers 1 and 2, then feeds those features into MioCodec's global encoder to produce the 128-dimensional speaker embedding used by the decoder.

Part of soniqo.audio, an on-device speech toolkit.

Model

FieldValue
Base modelmicrosoft/wavlm-base-plus
ArchitectureWavLM / HuBERT-style SSL encoder
Parameters94M class
FormatMLX-compatible safetensors
QuantizationNone
Precisionfp16-compatible safetensors export
Sample rate16 kHz
Hidden size768
Layers12 Transformer layers
Attention heads12
Runtime useIndic-Mio raw-reference speaker embedding companion

Files

FileSizeDescription
model.safetensors360 MBWavLM weights in safetensors format
config.json2.2 KBWavLM model configuration
preprocessor_config.json215 B16 kHz audio feature-extractor metadata
soniqo_manifest.json622 BRuntime/export metadata

Performance

CheckResult
Weight loadPasses in speech-swift WavLM companion loader
Indic-Mio raw-reference embeddingProduces finite 128-dimensional MioCodec global embedding
Indic-Mio raw-reference synthesisPasses local E2E synthesis test
Hindi ASR sanity on generated clone sampleRecovered नमस्ते यह संदर्भा आवाज़

Usage

Swift

swift
import IndicMioTTS

let model = try await IndicMioTTSModel.fromPretrained()
let audio = try await model.generate(
    text: "नमस्ते, यह संदर्भ आवाज़ का परीक्षण है। <happy>",
    language: "hindi",
    referenceAudio: referenceSamples,
    referenceSampleRate: referenceSampleRate
)

The runtime downloads this companion automatically when raw reference audio is used. For local testing with a pre-downloaded bundle:

bash
export INDIC_MIO_WAVLM_BUNDLE=/path/to/WavLM-Base-Plus-MLX-fp16

CLI

bash
speech speak \
  --engine indic-mio \
  --voice-sample reference.wav \
  --output clone.wav \
  "नमस्ते, यह संदर्भ आवाज़ का परीक्षण है। <happy>"

Source

Converted from microsoft/wavlm-base-plus. The upstream WavLM source project is available from microsoft/unilm.

Links

License

MIT, following the upstream WavLM source project license.