create links for embeddings

ddPn08 · Dec 30, 2022 · 869d6b8 · 869d6b8
1 parent 092888a
commit 869d6b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/automatic1111.ipynb b/automatic1111.ipynb
@@ -150,7 +150,10 @@
         "os.makedirs(f\"{models_path}/Stable-diffusion\", exist_ok=True)\n",
         "\n",
         "for dir in os.listdir(models_path):\n",
-        "    ! rm -Rf stable-diffusion-webui/models/{dir} && ln -s {models_path}/{dir} stable-diffusion-webui/models/{dir}\n",
+        "    if dir == \"embeddings\":\n",
+        "        ! rm -Rf stable-diffusion-webui/embeddings && ln -s {models_path}/embeddings stable-diffusion-webui/models/embeddings\n",
+        "    else:\n",
+        "        ! rm -Rf stable-diffusion-webui/models/{dir} && ln -s {models_path}/{dir} stable-diffusion-webui/models/{dir}\n",
         "\n",
         "! rm -f stable-diffusion-webui/outputs && ln -s {data_dir}/outputs stable-diffusion-webui/outputs\n",
         "! rm -f stable-diffusion-webui/config.json && ln -s {config_path}/config.json stable-diffusion-webui/config.json\n",