CoolFace
Modelpublic

aufklarer/WeSpeaker-ResNet34-LM-CoreML

sourceHugging Facemitupdated 5mo agoView on Hugging Face
0likes2.1kdownloads
Model Card

WeSpeaker-ResNet34-LM — CoreML

CoreML conversion of WeSpeaker ResNet34-LM for Apple Neural Engine.

Produces 256-dimensional L2-normalized speaker embeddings from audio.

Model Details

DetailValue
ArchitectureResNet34 with statistics pooling
Parameters~6.6M
Input80-bin log-mel spectrogram (16kHz)
Output256-dim L2-normalized speaker embedding
BatchNormFused into Conv2d at conversion time

Usage

swift
let model = try await WeSpeakerModel.fromPretrained(backend: .coreML)
let embedding = model.embed(audio: samples, sampleRate: 16000)
let similarity = WeSpeakerModel.cosineSimilarity(embeddingA, embeddingB)

Variants

VariantBackendModel ID
MLXGPUaufklarer/WeSpeaker-ResNet34-LM-MLX
CoreMLNeural Engineaufklarer/WeSpeaker-ResNet34-LM-CoreML

Links