Update Stable_Diffusion_AI_Notebook.ipynb

Making Stable_Diffusion_AI_Notebook.ipynb work smoothly on Google Colab

Author: ChloeL19

notebooks/Stable_Diffusion_AI_Notebook.ipynb

@@ -58,7 +58,7 @@
         "from os.path import exists\n",
         "\n",
         "!git clone --quiet https://github.com/invoke-ai/InvokeAI.git # Original repo\n",
-        "%cd /content/stable-diffusion/\n",
+        "%cd /content/InvokeAI/\n",
         "!git checkout --quiet tags/release-1.14.1"
       ]
     },
@@ -79,6 +79,7 @@
         "!pip install colab-xterm\n",
         "!pip install -r requirements-lin-win-colab-CUDA.txt\n",
         "!pip install clean-fid torchtext\n",
+        "!pip install transformers\n",
         "gc.collect()"
       ]
     },
@@ -106,7 +107,7 @@
       "source": [
         "#@title 5. Load small ML models required\n",
         "import gc\n",
-        "%cd /content/stable-diffusion/\n",
+        "%cd /content/InvokeAI/\n",
         "!python scripts/preload_models.py\n",
         "gc.collect()"
       ]
@@ -171,18 +172,18 @@
         "import os \n",
         "\n",
         "# Folder creation if it doesn't exist\n",
-        "if exists(\"/content/stable-diffusion/models/ldm/stable-diffusion-v1\"):\n",
+        "if exists(\"/content/InvokeAI/models/ldm/stable-diffusion-v1\"):\n",
         "     print(\"❗ Dir stable-diffusion-v1 already exists\")\n",
         "else:\n",
-        "    %mkdir /content/stable-diffusion/models/ldm/stable-diffusion-v1\n",
+        "    %mkdir /content/InvokeAI/models/ldm/stable-diffusion-v1\n",
         "    print(\"✅ Dir stable-diffusion-v1 created\")\n",
         "\n",
         "# Symbolic link if it doesn't exist\n",
-        "if exists(\"/content/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt\"):\n",
+        "if exists(\"/content/InvokeAI/models/ldm/stable-diffusion-v1/model.ckpt\"):\n",
         "    print(\"❗ Symlink already created\")\n",
         "else: \n",
         "    src = model_path\n",
-        "    dst = '/content/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt'\n",
+        "    dst = '/content/InvokeAI/models/ldm/stable-diffusion-v1/model.ckpt'\n",
         "    os.symlink(src, dst) \n",
         "    print(\"✅ Symbolic link created successfully\")"
       ]
@@ -233,7 +234,7 @@
         "%matplotlib inline\n",
         "\n",
         "images = []\n",
-        "for img_path in sorted(glob.glob('/content/stable-diffusion/outputs/img-samples/*.png'), reverse=True):\n",
+        "for img_path in sorted(glob.glob('/content/InvokeAI/outputs/img-samples/*.png'), reverse=True):\n",
         "    images.append(mpimg.imread(img_path))\n",
         "\n",
         "images = images[:15] \n",