Update README.md (#5)
Browse files- Update README.md (0f62d97957a239ceb5284c599f7ec6948620552d)
Co-authored-by: Xuesong Yang <magicyoung8@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -147,6 +147,8 @@ from nemo.collections.asr.models import SortformerEncLabelModel
|
|
| 147 |
diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
|
| 148 |
# load model from Hugging Face model card directly (You need a Hugging Face token)
|
| 149 |
diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
|
|
|
|
|
|
|
| 150 |
```
|
| 151 |
|
| 152 |
### Input Format
|
|
|
|
| 147 |
diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
|
| 148 |
# load model from Hugging Face model card directly (You need a Hugging Face token)
|
| 149 |
diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
|
| 150 |
+
# switch to inference mode
|
| 151 |
+
diar_model.eval()
|
| 152 |
```
|
| 153 |
|
| 154 |
### Input Format
|