Automatic Speech Recognition
NeMo
GGUF
PyTorch
speaker-diarization
speaker-recognition
speech
audio
Transformer
FastConformer
Conformer
NEST
NeMo
Eval Results (legacy)
Instructions to use nvidia/diar_streaming_sortformer_4spk-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use nvidia/diar_streaming_sortformer_4spk-v2 with NeMo:
import nemo.collections.asr as nemo_asr asr_model = nemo_asr.models.ASRModel.from_pretrained("nvidia/diar_streaming_sortformer_4spk-v2") transcriptions = asr_model.transcribe(["file.wav"]) - Notebooks
- Google Colab
- Kaggle
add: q8 GGUF for local inference with NeMo-Speech.cpp
#13
by prabhsimrans-nv - opened
- .gitattributes +1 -0
- README.md +32 -1
- diar_streaming_sortformer_4spk-v2.q8_0.gguf +3 -0
.gitattributes
CHANGED
|
@@ -33,6 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 36 |
fifo.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
streaming_steps.png filter=lfs diff=lfs merge=lfs -text
|
| 38 |
sortformer_intro.png filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
*.gguf filter=lfs diff=lfs merge=lfs -text
|
| 37 |
fifo.png filter=lfs diff=lfs merge=lfs -text
|
| 38 |
streaming_steps.png filter=lfs diff=lfs merge=lfs -text
|
| 39 |
sortformer_intro.png filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -267,6 +267,37 @@ for segment in predicted_segments[0]:
|
|
| 267 |
```
|
| 268 |
## How to Use this Model
|
| 269 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 270 |
The model is available for use in the NeMo Framework[6], and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.
|
| 271 |
|
| 272 |
### Loading the Model
|
|
@@ -506,4 +537,4 @@ Also check out the [Riva live demo](https://developer.nvidia.com/riva#demos).
|
|
| 506 |
|
| 507 |
## Licence
|
| 508 |
|
| 509 |
-
License to use this model is covered by the [CC-BY-4.0](https://creativecommons.org/licenses/by/4.0/legalcode). By downloading the public and release version of the model, you accept the terms and conditions of the CC-BY-4.0 license.
|
|
|
|
| 267 |
```
|
| 268 |
## How to Use this Model
|
| 269 |
|
| 270 |
+
There are several ways to use this model. Choose the one that fits your needs.
|
| 271 |
+
|
| 272 |
+
### Run locally with NeMo-Speech.cpp
|
| 273 |
+
|
| 274 |
+
[NeMo-Speech.cpp](https://github.com/NVIDIA/NeMo-Speech.cpp) provides a
|
| 275 |
+
lightweight native C++ runtime for local speaker diarization. After
|
| 276 |
+
[installing the runtime](https://github.com/NVIDIA/NeMo-Speech.cpp#installation):
|
| 277 |
+
|
| 278 |
+
```bash
|
| 279 |
+
hf download nvidia/diar_streaming_sortformer_4spk-v2 \
|
| 280 |
+
diar_streaming_sortformer_4spk-v2.q8_0.gguf \
|
| 281 |
+
--local-dir models
|
| 282 |
+
|
| 283 |
+
nemo-speech diarize meeting.wav \
|
| 284 |
+
--model models/diar_streaming_sortformer_4spk-v2.q8_0.gguf
|
| 285 |
+
```
|
| 286 |
+
|
| 287 |
+
The same model can add word-level speaker tags to a transcription:
|
| 288 |
+
|
| 289 |
+
```bash
|
| 290 |
+
nemo-speech transcribe meeting.wav \
|
| 291 |
+
--model models/asr-model.gguf \
|
| 292 |
+
--diar-model models/diar_streaming_sortformer_4spk-v2.q8_0.gguf \
|
| 293 |
+
--json
|
| 294 |
+
```
|
| 295 |
+
|
| 296 |
+
See the [NeMo-Speech.cpp diarization guide](https://github.com/NVIDIA/NeMo-Speech.cpp/blob/main/docs/cli.md#diarize-audio)
|
| 297 |
+
for more usage examples.
|
| 298 |
+
|
| 299 |
+
### NVIDIA NeMo
|
| 300 |
+
|
| 301 |
The model is available for use in the NeMo Framework[6], and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.
|
| 302 |
|
| 303 |
### Loading the Model
|
|
|
|
| 537 |
|
| 538 |
## Licence
|
| 539 |
|
| 540 |
+
License to use this model is covered by the [CC-BY-4.0](https://creativecommons.org/licenses/by/4.0/legalcode). By downloading the public and release version of the model, you accept the terms and conditions of the CC-BY-4.0 license.
|
diar_streaming_sortformer_4spk-v2.q8_0.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0679cfeb1ce356d0dea9470b31274f4bfc7eb927497d82005483770666da998a
|
| 3 |
+
size 147075776
|