Add Transformers weights, config and usage section
#1
by eustlb HF Staff - opened
Adds the 🤗 Transformers weights and configuration for this checkpoint, converted from Nemotron-3-Diarization.nemo with the conversion script in the Transformers PR, plus a Transformers usage section in README.md (offline and streaming diarization).
The auxiliary activity head (only used by the training loss) is not converted. Weights are stored in float32, matching NeMo's inference precision.
Draft: Transformers support for this architecture is not merged yet, so the weight names and the configuration schema may still change during review. The snippets install Transformers from source for that reason. I will refresh this PR if review changes the layout.
eustlb changed pull request status to open
taejinp changed pull request status to merged