update branch

Signed-off-by: ericharper <[email protected]>
anmolgupt · Apr 11, 2023 · d9650a5 · d9650a5
1 parent 159b9a8
commit d9650a5
Show file tree

Hide file tree

Showing 64 changed files with 227 additions and 227 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
diff --git a/tutorials/00_NeMo_Primer.ipynb b/tutorials/00_NeMo_Primer.ipynb
@@ -42,7 +42,7 @@
         "!pip install text-unidecode\n",
         "\n",
         "# ## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Install TorchAudio\n",

diff --git a/tutorials/01_NeMo_Models.ipynb b/tutorials/01_NeMo_Models.ipynb
@@ -37,7 +37,7 @@
         "!pip install text-unidecode\n",
         "\n",
         "# ## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Install TorchAudio\n",

diff --git a/tutorials/02_NeMo_Adapters.ipynb b/tutorials/02_NeMo_Adapters.ipynb
@@ -25,7 +25,7 @@
         "!pip install text-unidecode\n",
         "\n",
         "# ## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Grab the config we'll use in this example\n",

diff --git a/tutorials/AudioTranslationSample.ipynb b/tutorials/AudioTranslationSample.ipynb
@@ -38,7 +38,7 @@
             },
             "outputs": [],
             "source": [
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n"
             ]
         },

diff --git a/tutorials/Publish_NeMo_Model_On_Hugging_Face_Hub.ipynb b/tutorials/Publish_NeMo_Model_On_Hugging_Face_Hub.ipynb
@@ -41,7 +41,7 @@
         "!pip install text-unidecode\n",
         "\n",
         "### Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]"
       ]
     },

diff --git a/tutorials/asr/ASR_CTC_Language_Finetuning.ipynb b/tutorials/asr/ASR_CTC_Language_Finetuning.ipynb
@@ -39,7 +39,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "\"\"\"\n",

diff --git a/tutorials/asr/ASR_for_telephony_speech.ipynb b/tutorials/asr/ASR_for_telephony_speech.ipynb
@@ -27,7 +27,7 @@
                 "!pip install matplotlib>=3.3.2\n",
                 "\n",
                 "## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
                 "\n",
                 "## Grab the config we'll use in this example\n",

diff --git a/tutorials/asr/ASR_with_NeMo.ipynb b/tutorials/asr/ASR_with_NeMo.ipynb
@@ -53,7 +53,7 @@
                 "!pip install matplotlib>=3.3.2\n",
                 "\n",
                 "## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
                 "\n",
                 "\"\"\"\n",
@@ -587,7 +587,7 @@
                 "\n",
                 "if not os.path.exists(config_path):\n",
                 "    # Grab the config we'll use in this example\n",
-                "    BRANCH = 'main'\n",
+                "    BRANCH = 'r1.17.0_pt_23.04'\n",
                 "    !mkdir configs\n",
                 "    !wget -P configs/ https://raw.githubusercontent.com/NVIDIA/NeMo/$BRANCH/examples/asr/conf/config.yaml\n",
                 "\n",

diff --git a/tutorials/asr/ASR_with_Subword_Tokenization.ipynb b/tutorials/asr/ASR_with_Subword_Tokenization.ipynb
@@ -40,7 +40,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Grab the config we'll use in this example\n",

diff --git a/tutorials/asr/ASR_with_Transducers.ipynb b/tutorials/asr/ASR_with_Transducers.ipynb
@@ -28,7 +28,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Grab the config we'll use in this example\n",

diff --git a/tutorials/asr/Buffered_Transducer_Inference.ipynb b/tutorials/asr/Buffered_Transducer_Inference.ipynb
@@ -27,7 +27,7 @@
     "!pip install matplotlib>=3.3.2\n",
     "\n",
     "## Install NeMo\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "# Update numba and restart (this is required to update internal numba version of Colab)\n",

diff --git a/tutorials/asr/Buffered_Transducer_Inference_with_LCS_Merge.ipynb b/tutorials/asr/Buffered_Transducer_Inference_with_LCS_Merge.ipynb
@@ -45,7 +45,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "# Update numba and restart (this is required to update internal numba version of Colab)\n",

diff --git a/tutorials/asr/Intro_to_Transducers.ipynb b/tutorials/asr/Intro_to_Transducers.ipynb
@@ -43,7 +43,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]"
       ],
       "execution_count": null,

diff --git a/tutorials/asr/Multilang_ASR.ipynb b/tutorials/asr/Multilang_ASR.ipynb
@@ -101,7 +101,7 @@
     "\n",
     "## Install NeMo\n",
     "## We are using the main branch but you might want to adjust that too\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "\"\"\"\n",

diff --git a/tutorials/asr/Offline_ASR.ipynb b/tutorials/asr/Offline_ASR.ipynb
@@ -51,7 +51,7 @@
         "id": "I9eIxAyKHREB"
       },
       "source": [
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "try:\n",
         "    # Import NeMo Speech Recognition collection\n",
         "    import nemo.collections.asr as nemo_asr\n",

diff --git a/tutorials/asr/Offline_ASR_with_VAD_for_CTC_models.ipynb b/tutorials/asr/Offline_ASR_with_VAD_for_CTC_models.ipynb
@@ -22,7 +22,7 @@
                 "!pip install wget\n",
                 "\n",
                 "## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
                 "\n",
                 "\"\"\"\n",

diff --git a/tutorials/asr/Online_ASR_Microphone_Demo.ipynb b/tutorials/asr/Online_ASR_Microphone_Demo.ipynb
@@ -26,7 +26,7 @@
                 "!pip install pyaudio\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Grab the config we'll use in this example\n",

diff --git a/tutorials/asr/Online_Noise_Augmentation.ipynb b/tutorials/asr/Online_Noise_Augmentation.ipynb
@@ -31,7 +31,7 @@
                 "!pip install text-unidecode\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Install TorchAudio\n",

diff --git a/tutorials/asr/Online_Offline_Microphone_VAD_Demo.ipynb b/tutorials/asr/Online_Offline_Microphone_VAD_Demo.ipynb
@@ -26,7 +26,7 @@
                 "!pip install pyaudio\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Install TorchAudio\n",

diff --git a/tutorials/asr/Online_Offline_Speech_Commands_Demo.ipynb b/tutorials/asr/Online_Offline_Speech_Commands_Demo.ipynb
@@ -28,7 +28,7 @@
                 "!pip install pyaudio\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Install TorchAudio\n",

diff --git a/tutorials/asr/Self_Supervised_Pre_Training.ipynb b/tutorials/asr/Self_Supervised_Pre_Training.ipynb
@@ -27,7 +27,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "\"\"\"\n",

diff --git a/tutorials/asr/Speech_Commands.ipynb b/tutorials/asr/Speech_Commands.ipynb
@@ -60,7 +60,7 @@
                 "!pip install text-unidecode\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Install TorchAudio\n",

diff --git a/tutorials/asr/Streaming_ASR.ipynb b/tutorials/asr/Streaming_ASR.ipynb
@@ -27,7 +27,7 @@
     "!pip install matplotlib>=3.3.2\n",
     "\n",
     "## Install NeMo\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "## Grab the config we'll use in this example\n",

diff --git a/tutorials/asr/Voice_Activity_Detection.ipynb b/tutorials/asr/Voice_Activity_Detection.ipynb
@@ -27,7 +27,7 @@
                 "!pip install text-unidecode\n",
                 "\n",
                 "# ## Install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[asr]\n",
                 "\n",
                 "## Install TorchAudio\n",

diff --git a/tutorials/asr/asr_adapters/ASR_with_Adapters.ipynb b/tutorials/asr/asr_adapters/ASR_with_Adapters.ipynb
@@ -49,7 +49,7 @@
         "!pip install matplotlib>=3.3.2\n",
         "\n",
         "## Install NeMo\n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
         "\n",
         "## Grab the config we'll use in this example\n",

diff --git a/tutorials/cloud/aws/ASR_Finetuning_at_Scale_with_AWS_SageMaker.ipynb b/tutorials/cloud/aws/ASR_Finetuning_at_Scale_with_AWS_SageMaker.ipynb
@@ -70,7 +70,7 @@
     "!pip install matplotlib>=3.3.2\n",
     "\n",
     "## Install NeMo\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "\"\"\"\n",

diff --git a/tutorials/cloud/aws/SageMaker_ASR_Training.ipynb b/tutorials/cloud/aws/SageMaker_ASR_Training.ipynb
@@ -55,7 +55,7 @@
     "!pip install matplotlib>=3.3.2\n",
     "\n",
     "## Install NeMo\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "\"\"\"\n",

diff --git a/tutorials/nlp/01_Pretrained_Language_Models_for_Downstream_Tasks.ipynb b/tutorials/nlp/01_Pretrained_Language_Models_for_Downstream_Tasks.ipynb
@@ -26,7 +26,7 @@
                 "# If you're using Google Colab and not running locally, run this cell\n",
                 "\n",
                 "# install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]"
             ]
         },

diff --git a/tutorials/nlp/02_NLP_Tokenizers.ipynb b/tutorials/nlp/02_NLP_Tokenizers.ipynb
@@ -10,7 +10,7 @@
             },
             "outputs": [],
             "source": [
-                "BRANCH = 'main'"
+                "BRANCH = 'r1.17.0_pt_23.04'"
             ]
         },
         {
@@ -35,7 +35,7 @@
                 "# If you're using Google Colab and not running locally, run this cell\n",
                 "\n",
                 "# install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]"
             ]
         },

diff --git a/tutorials/nlp/Data_Preprocessing_and_Cleaning_for_NMT.ipynb b/tutorials/nlp/Data_Preprocessing_and_Cleaning_for_NMT.ipynb
@@ -300,7 +300,7 @@
     "\n",
     "## Install NeMo\n",
     "\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]\n",
     "\n",
     "!pip uninstall -y sacrebleu\n",

diff --git a/tutorials/nlp/Dialogue.ipynb b/tutorials/nlp/Dialogue.ipynb
@@ -27,7 +27,7 @@
       "outputs": [],
       "source": [
         "import os \n",
-        "BRANCH = 'main'\n",
+        "BRANCH = 'r1.17.0_pt_23.04'\n",
         "!apt-get update && apt-get install -y libsndfile1 ffmpeg\n",
         "!git clone https://github.com/NVIDIA/NeMo --branch $BRANCH\n",
         "os.chdir('NeMo')\n",

diff --git a/tutorials/nlp/Entity_Linking_Medical.ipynb b/tutorials/nlp/Entity_Linking_Medical.ipynb
@@ -17,7 +17,7 @@
     "\"\"\"\n",
     "\n",
     "## Install NeMo if using google collab or if its not installed locally\n",
-    "BRANCH = 'main'\n",
+    "BRANCH = 'r1.17.0_pt_23.04'\n",
     "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]"
    ]
   },

diff --git a/tutorials/nlp/GLUE_Benchmark.ipynb b/tutorials/nlp/GLUE_Benchmark.ipynb
@@ -44,7 +44,7 @@
         "# If you're using Google Colab and not running locally, run this cell\n",
         "\n",
         "# install NeMo\n",
-        "BRANCH = 'main'\n!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]\n"
+        "BRANCH = 'r1.17.0_pt_23.04'\n!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]\n"
       ],
       "execution_count": null,
       "outputs": []

diff --git a/tutorials/nlp/Joint_Intent_and_Slot_Classification.ipynb b/tutorials/nlp/Joint_Intent_and_Slot_Classification.ipynb
@@ -22,7 +22,7 @@
                 "# If you're using Google Colab and not running locally, run this cell\n",
                 "\n",
                 "# install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]"
             ]
         },

diff --git a/tutorials/nlp/MegatronBert_export.ipynb b/tutorials/nlp/MegatronBert_export.ipynb
@@ -7,7 +7,7 @@
             "metadata": {},
             "outputs": [],
             "source": [
-                "BRANCH='main'"
+                "BRANCH='r1.17.0_pt_23.04'"
             ]
         },
         {

diff --git a/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb b/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb
@@ -62,7 +62,7 @@
             "metadata": {},
             "outputs": [],
             "source": [
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "DATA_PATH='.'\n",
                 "TRANSACTIONS=DATA_PATH+'/card_transaction.v1.csv'\n",
                 "#CHECKPOINTS='/chk_points'\n",

diff --git a/tutorials/nlp/Multitask_Prompt_and_PTuning.ipynb b/tutorials/nlp/Multitask_Prompt_and_PTuning.ipynb
@@ -7,7 +7,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "BRANCH='main'"
+    "BRANCH='r1.17.0_pt_23.04'"
    ]
   },
   {

diff --git a/tutorials/nlp/Punctuation_and_Capitalization.ipynb b/tutorials/nlp/Punctuation_and_Capitalization.ipynb
@@ -6,7 +6,7 @@
             "metadata": {},
             "outputs": [],
             "source": [
-                "BRANCH = 'main'"
+                "BRANCH = 'r1.17.0_pt_23.04'"
             ]
         },
         {

diff --git a/tutorials/nlp/Punctuation_and_Capitalization_Lexical_Audio.ipynb b/tutorials/nlp/Punctuation_and_Capitalization_Lexical_Audio.ipynb
@@ -10,7 +10,7 @@
    },
    "outputs": [],
    "source": [
-    "BRANCH = 'main'"
+    "BRANCH = 'r1.17.0_pt_23.04'"
    ]
   },
   {

diff --git a/tutorials/nlp/Question_Answering.ipynb b/tutorials/nlp/Question_Answering.ipynb
@@ -74,7 +74,7 @@
       },
       "outputs": [],
       "source": [
-        "BRANCH = 'main'"
+        "BRANCH = 'r1.17.0_pt_23.04'"
       ]
     },
     {

diff --git a/tutorials/nlp/Relation_Extraction-BioMegatron.ipynb b/tutorials/nlp/Relation_Extraction-BioMegatron.ipynb
@@ -6,7 +6,7 @@
             "metadata": {},
             "outputs": [],
             "source": [
-                "BRANCH = 'main'"
+                "BRANCH = 'r1.17.0_pt_23.04'"
             ]
         },
         {

diff --git a/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb b/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb
@@ -20,7 +20,7 @@
                 "# If you're using Google Colab and not running locally, run this cell\n",
                 "\n",
                 "# install NeMo\n",
-                "BRANCH = 'main'\n",
+                "BRANCH = 'r1.17.0_pt_23.04'\n",
                 "!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[nlp]\n",
                 "\n"
             ]