Checkout our HuggingFace🤗 collection for the latest open weight checkpoints and demos!
NVIDIA NeMo Speech is built for researchers and PyTorch developers working on Speech models including Automatic Speech Recognition (ASR), Text to Speech (TTS), and Speech LLMs. It is designed to help you efficiently create, customize, and deploy new It is designed to help you efficiently create, customize, and deploy new AI models by leveraging existing code and pre-trained model checkpoints.
For technical documentation, please see the NeMo Framework User Guide.
As of Pytorch 2.6,
torch.load defaults to using weights_only=True. Some model checkpoints may require using weights_only=False.
In this case, you can set the env var TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1 before running code that uses torch.load.
However, this should only be done with trusted files. Loading files from untrusted sources with more than weights only
can have the risk of arbitrary code execution.
| Version | Status | Description |
|---|---|---|
| Latest | Documentation of the latest (i.e. main) branch. | |
| Stable | Documentation of the stable (i.e. most recent release) - To be added |
NeMo Speech is installable via pip: pip install 'nemo-toolkit[all]'
To install with extra dependencies for CUDA 12.x or 13.x, use pip install 'nemo-toolkit[all,cu12]'
or pip install 'nemo-toolkit[all,cu13]' respectively.
We welcome community contributions! Please refer to CONTRIBUTING.md for the process.
NeMo is licensed under the Apache License 2.0.
Checkout our HuggingFace🤗 collection for the latest open weight checkpoints and demos!
NVIDIA NeMo Speech is built for researchers and PyTorch developers working on Speech models including Automatic Speech Recognition (ASR), Text to Speech (TTS), and Speech LLMs. It is designed to help you efficiently create, customize, and deploy new It is designed to help you efficiently create, customize, and deploy new AI models by leveraging existing code and pre-trained model checkpoints.
For technical documentation, please see the NeMo Framework User Guide.
As of Pytorch 2.6,
torch.load defaults to using weights_only=True. Some model checkpoints may require using weights_only=False.
In this case, you can set the env var TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1 before running code that uses torch.load.
However, this should only be done with trusted files. Loading files from untrusted sources with more than weights only
can have the risk of arbitrary code execution.
| Version | Status | Description |
|---|---|---|
| Latest | Documentation of the latest (i.e. main) branch. | |
| Stable | Documentation of the stable (i.e. most recent release) - To be added |
NeMo Speech is installable via pip: pip install 'nemo-toolkit[all]'
To install with extra dependencies for CUDA 12.x or 13.x, use pip install 'nemo-toolkit[all,cu12]'
or pip install 'nemo-toolkit[all,cu13]' respectively.
We welcome community contributions! Please refer to CONTRIBUTING.md for the process.
NeMo is licensed under the Apache License 2.0.