diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 9f8a62b..0fb2577 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -354,7 +354,7 @@
         "if Disable_text_Encoder:\n",
         "  textencode = \"--disable_textenc_training Train_text \"\n",
         "\n",
-        "!python train_colab.py --resume_ckpt \"$model\" \\\n",
+        "!python train.py --resume_ckpt \"$model\" \\\n",
         "  $textencode \\\n",
         "  $Gradient \\\n",
         "  $shuffle \\\n",
@@ -378,8 +378,8 @@
         "  --save_every_n_epoch $Save_every_N_epoch \\\n",
         "  --seed 555 \\\n",
         "  --shuffle_tags \\\n",
-        "  --useadam8bit \\\n",
-        "  --notebook\n"
+        "  --useadam8bit \n",
+
       ]
     },
     {
@@ -464,4 +464,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}