Integrating Conversational AI Models into Riva
While the Transfer Learning Toolkit is an excellent resource to train and finetune models, Riva provides resources to deploy those models into scalable services running on GPUs.
Along with the TLT conversational AI package, we provide the following sample resources on NGC to capture the end to end workflow of training a model with TLT and deploying them to Riva.
Conversational AI Task |
Jupyter Notebooks |
---|---|
Speech to Text |
|
Speech to Text Citrinet |
|
Question Answering |
|
Text Classification |
|
Token Classification |
|
Punctuation and Capitalization |
|
Intent and Slot Classification |
Each sample resouce contain 2 sample notebooks,
To train the respective model using TLT and generate an exported
.ejrvs
To use this exported
.ejrvs
file and deploy it to Riva.
You may find more information about the same in the Riva Documentation.