train.py

now works with colab and colab can use also make use of tqdm.auto so --notebook is unneeded

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
-}
+}