From 5c343c73c6cba2a11eec8fe13bbf44b0cfa08a1b Mon Sep 17 00:00:00 2001
From: Yi Dong <yidong@nvidia.com>
Date: Thu, 8 Jun 2023 18:55:33 +0000
Subject: [PATCH] fix notebook error

Signed-off-by: Yi Dong <yidong@nvidia.com>
---
 ...on_Synthetic_Tabular_Data_Generation.ipynb | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb b/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb
index a92317b17320..1d9849467c21 100644
--- a/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb
+++ b/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb
@@ -1,6 +1,7 @@
 {
     "cells": [
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "c3217a15",
             "metadata": {},
@@ -15,6 +16,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "8c72dc42",
             "metadata": {},
@@ -25,6 +27,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "79154a9e",
             "metadata": {},
@@ -73,6 +76,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "7e0bbc89",
             "metadata": {},
@@ -92,6 +96,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "1ff1d46f",
             "metadata": {},
@@ -141,6 +146,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "aa356012",
             "metadata": {},
@@ -239,6 +245,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "02bff63f",
             "metadata": {},
@@ -267,6 +274,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "89e1e5b3",
             "metadata": {},
@@ -339,6 +347,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "05ebadc3",
             "metadata": {},
@@ -347,6 +356,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "2fe38a29",
             "metadata": {},
@@ -381,6 +391,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "678f65ef",
             "metadata": {},
@@ -411,6 +422,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "8af66b4a",
             "metadata": {},
@@ -464,6 +476,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "6ecec681",
             "metadata": {},
@@ -472,6 +485,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "58a3d4fa",
             "metadata": {},
@@ -543,6 +557,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "45ac928f",
             "metadata": {},
@@ -557,6 +572,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "158a4bbe",
             "metadata": {},
@@ -586,7 +602,7 @@
             "outputs": [],
             "source": [
                 "CHECKPONT_FILE_NAME = megatron_gpt--val_loss=1.17-step=10047-consumed_samples=80376.0-last.ckpt # change it to your checkpoint file name\n",
-                "!python -m torch.distributed.launch --nproc_per_node=1 megatron_ckpt_to_nemo.py \\\n",
+                "!python -m torch.distributed.launch --nproc_per_node=1 --use-env=True megatron_ckpt_to_nemo.py \\\n",
                 "    --checkpoint_folder=gpt_creditcard_results/megatron_gpt/checkpoints/ \\\n",
                 "    --checkpoint_name={CHECKPONT_FILE_NAME} \\\n",
                 "    --nemo_file_path=tabular.nemo \\\n",
@@ -597,6 +613,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "fa16378e",
             "metadata": {},
@@ -605,6 +622,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "ed056ec6",
             "metadata": {},
@@ -630,6 +648,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "a62b48dc",
             "metadata": {},
@@ -685,6 +704,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "cccd54d9",
             "metadata": {},
@@ -790,6 +810,7 @@
             ]
         },
         {
+            "attachments": {},
             "cell_type": "markdown",
             "id": "0f2f6e3a",
             "metadata": {},