Updated some missing docs about training and setting up riva

This commit is contained in:
Dave Niewinski 2023-12-11 12:05:45 -05:00
parent 1239c6735c
commit b8f2cda369
2 changed files with 53 additions and 23 deletions

1
TrainNemo/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
.ipynb_checkpoints

View File

@ -3,7 +3,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "b276bbe0",
"id": "90ce109a",
"metadata": {},
"outputs": [],
"source": [
@ -21,7 +21,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "a9bb6563",
"id": "84df2778",
"metadata": {},
"outputs": [],
"source": [
@ -52,7 +52,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "c3b7b7e3",
"id": "6c36b288",
"metadata": {},
"outputs": [],
"source": [
@ -199,7 +199,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "82a7d268",
"id": "8c883c18",
"metadata": {},
"outputs": [],
"source": [
@ -209,7 +209,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "832fcd67",
"id": "986e916e",
"metadata": {},
"outputs": [],
"source": [
@ -220,7 +220,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "e56749bf",
"id": "ae039452",
"metadata": {},
"outputs": [],
"source": [
@ -232,7 +232,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "9bf1fb18",
"id": "3717107a",
"metadata": {},
"outputs": [],
"source": [
@ -256,7 +256,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "aefc57ff",
"id": "185a6f9b",
"metadata": {},
"outputs": [],
"source": [
@ -273,7 +273,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "64b51e94",
"id": "1dd57280",
"metadata": {},
"outputs": [],
"source": [
@ -288,7 +288,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "6b92a469",
"id": "98da0076",
"metadata": {},
"outputs": [],
"source": [
@ -314,7 +314,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "780dba9a",
"id": "7ed81322",
"metadata": {},
"outputs": [],
"source": [
@ -328,7 +328,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "831239e2",
"id": "56a8945f",
"metadata": {},
"outputs": [],
"source": [
@ -384,7 +384,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "2fad0610",
"id": "7d6594e8",
"metadata": {},
"outputs": [],
"source": [
@ -419,7 +419,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "0454967c",
"id": "e36967ad",
"metadata": {},
"outputs": [],
"source": [
@ -509,7 +509,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "9d6dc1ee",
"id": "1178d4b7",
"metadata": {},
"outputs": [],
"source": [
@ -531,7 +531,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "45085a4e",
"id": "741f10e2",
"metadata": {},
"outputs": [],
"source": [
@ -577,7 +577,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "50b4c3be",
"id": "f75da634",
"metadata": {},
"outputs": [],
"source": [
@ -590,7 +590,7 @@
{
"cell_type": "code",
"execution_count": null,
"id": "584dc353",
"id": "4dad4058",
"metadata": {},
"outputs": [],
"source": [
@ -601,20 +601,49 @@
{
"cell_type": "code",
"execution_count": null,
"id": "52c6d0a9",
"id": "5d481c7b",
"metadata": {},
"outputs": [],
"source": [
"# ngc registry resource download-version nvidia/riva/riva_quickstart:2.8.1\n",
"# pip3 install nema2riva whl\n",
"# pip3 install whl\n",
"# pip3 install nemo2riva\n",
"\n",
"# Run on training machine\n",
"!nemo2riva --out hifigan.riva hifigan.nemo --key None\n",
"!nemo2riva --out fastpitch.riva fastpitch.nemo --key None"
"!nemo2riva --out fastpitch.riva fastpitch.nemo --key None\n",
"\n",
"# scp .riva files to jetson under /home/davesarmoury/RIVA/artifacts\n",
"# Everything below is run on the jetson\n",
"\n",
"# Servicemaker docker\n",
"docker run --gpus all -it --rm \\\n",
" -v /home/davesarmoury/RIVA/artifacts:/servicemaker-dev \\\n",
" -v /home/davesarmoury/RIVA/riva_repo:/data \\\n",
" --entrypoint=\\\"/bin/bash\\\" \\\n",
" nvcr.io/nvidia/riva/riva-speech:2.13.1-servicemaker-l4t-aarch64\n",
"\n",
"riva-build speech_synthesis \\\n",
" /servicemaker-dev/glados.rmir:tlt_encode \\\n",
" /servicemaker-dev/glados_fastpitch.riva:tlt_encode \\\n",
" /servicemaker-dev/glados_hifigan.riva:tlt_encode \\\n",
" --voice_name=GLaDOS \\\n",
" --sample_rate 22050\n",
"\n",
"riva-deploy /servicemaker-dev/glados.rmir:tlt_encode /data/models\n",
"\n",
"# Exit docker\n",
"\n",
"ngc registry resource download-version nvidia/riva/riva_quickstart\n",
"cd riva_quickstart\n",
"bash riva_init.sh\n",
"# copy glados.riva files into riva models dir\n",
"bash riva_start.sh"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"display_name": "Python 3",
"language": "python",
"name": "python3"
},