diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 0b50fa8..5158c25 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -114,6 +114,7 @@
         "clear_output()\n",
         "!git clone https://github.com/victorchall/EveryDream2trainer.git\n",
         "%cd /content/EveryDream2trainer\n",
+        "!python utils/get_yamls.py\n",
         "clear_output()\n",
         "print(\"DONE!\")"
       ]
@@ -134,7 +135,6 @@
         "from IPython.display import clear_output\n",
         "!mkdir input\n",
         "%cd /content/EveryDream2trainer\n",
-        "!python utils/get_yamls.py\n",
         "MODEL_URL = \"https://huggingface.co/panopstor/EveryDream/resolve/main/sd_v1-5_vae.ckpt\" #@param [\"https://huggingface.co/panopstor/EveryDream/resolve/main/sd_v1-5_vae.ckpt\", \"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float16.ckpt\"] {allow-input: true}\n",
         "print(\"Downloading \")\n",
         "!wget $MODEL_URL\n",