sortformer

Model type	sortformer
Class	SortformerDiarizationModel
Task	diarization
Source	models/sortformer.py

Description

Streaming Sortformer speaker-diarization model (offline forward).

Replicates NVIDIA NeMo’s SortformerEncLabelModel offline inference: a FastConformer encoder followed by a Transformer encoder and a speaker sigmoid head. Consumes mel-spectrogram features and returns per-frame speaker-activity probabilities in [0, 1].

Usage

mobius build --model <MODEL_ID> output_dir/

from mobius import build

model = build("<MODEL_ID>")