Updated license and model loading
Browse files
README.md
CHANGED
@@ -143,8 +143,10 @@ The model is available for use in the NeMo Framework[5], and can be used as a pr
|
|
143 |
```python
|
144 |
from nemo.collections.asr.models import SortformerEncLabelModel
|
145 |
|
146 |
-
# load model
|
147 |
diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
|
|
|
|
|
148 |
```
|
149 |
|
150 |
### Input Format
|
@@ -322,4 +324,4 @@ Check out [Riva live demo](https://developer.nvidia.com/riva#demos).
|
|
322 |
|
323 |
## Licence
|
324 |
|
325 |
-
License to use this model is covered by the [CC-BY-NC-
|
|
|
143 |
```python
|
144 |
from nemo.collections.asr.models import SortformerEncLabelModel
|
145 |
|
146 |
+
# load model from a downloaded file
|
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
|
|
|
324 |
|
325 |
## Licence
|
326 |
|
327 |
+
License to use this model is covered by the [CC-BY-NC-4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode). By downloading the public and release version of the model, you accept the terms and conditions of the CC-BY-NC-SA-4.0 license.
|