StephennFernandes's picture
Anuvani-Diarization 1.0.0
63dd946
Raw History Blame Contribute Delete
623 Bytes
# Copyright (c) 2026 The Anuvani-Diarization authors
# SPDX-License-Identifier: MIT OR Apache-2.0
"""The segmentation network: DiariZen's pruned WavLM-Large + Conformer powerset segmenter, without pyannote.
wav2vec2/, conformer.py and wavlm_config.py are verbatim copies from DiariZen (commit 844f555, MIT licence, see
../LICENSES/DiariZen-MIT.txt), which derive from torchaudio and DPHuBERT. segnet.py re-states DiariZen's
Model.__init__/forward without the pyannote base class; it is bit-identical to the release's network (all batch shapes,
assets/nn_golden.json).
"""
from .segnet import SegmentationNet # noqa: F401