Update VAE settings

acheong08 · Oct 25, 2022 · 0e3ac14 · 0e3ac14
1 parent 73111fe
commit 0e3ac14
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/Diffusion_WebUI.ipynb b/Diffusion_WebUI.ipynb
@@ -6,7 +6,7 @@
       "provenance": [],
       "collapsed_sections": [],
       "private_outputs": true,
-      "authorship_tag": "ABX9TyMQtq259OI+0zihoDH0F52W",
+      "authorship_tag": "ABX9TyPEcO2bHj27v3uSSC/TWYQw",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -135,14 +135,12 @@
         "  !git clone https://github.com/acheong08/stable-diffusion-webui\n",
         "  #@markdown Choose the models you want\n",
         "  use_hypernetworks = False #@param {'type':'boolean'}\n",
-        "  NovelAI = False #@param {'type':'boolean'}\n",
+        "  NovelAI = True #@param {'type':'boolean'}\n",
         "  Stable_Diffusion = False #@param {'type':'boolean'}\n",
         "  Waifu_Diffusion = False #@param {'type':'boolean'}\n",
         "  H_Diffusion = False #@param {'type':'boolean'}\n",
-        "  SD_V1_5 = True #@param {'type':'boolean'}\n",
+        "  SD_V1_5 = False #@param {'type':'boolean'}\n",
         "  SD_v1_5_inpainting = False #@param {'type':'boolean'}\n",
-        "  animeVae = False #@param {'type':'boolean'}\n",
-        "  SDVae = False #@param {'type':'boolean'}\n",
         "  if NovelAI:\n",
         "    custom_model(\"https://huggingface.co/acheong08/secretAI/resolve/main/stableckpt/animefull-final-pruned/model.ckpt\", \"novelAI\")\n",
         "  if Stable_Diffusion:\n",
@@ -157,13 +155,6 @@
         "    custom_model(\"https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt\", \"sd-v1-5-inpainting\")\n",
         "  if use_hypernetworks:\n",
         "    get_hypernetworks()\n",
-        "  if animeVae:\n",
-        "    !wget -c https://huggingface.co/acheong08/secretAI/resolve/main/stableckpt/animevae.pt -O {root_dir}/stable-diffusion-webui/models/Stable-diffusion/novelAI.vae.pt\n",
-        "    vae_args = \"--vae-path {root_dir}/stable-diffusion-webui/models/Stable-diffusion/novelAI.vae.pt\"\n",
-        "  if SDVae:\n",
-        "    !wget -c https://huggingface.co/stabilityai/sd-vae-ft-mse/resolve/main/diffusion_pytorch_model.bin -O {root_dir}/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5.vae.pt\n",
-        "    vae_args = \"--vae-path {root_dir}/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5.vae.pt\"\n",
-        "\n",
         "    \n",
         "  %cd {root_dir}/stable-diffusion-webui/extensions\n",
         "  !git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser\n",
@@ -196,6 +187,15 @@
         "    %pip install -q https://github.com/TheLastBen/fast-stable-diffusion/raw/main/precompiled/A100/xformers-0.0.13.dev0-py3-none-any.whl\n",
         "\n",
         "def run_webui():\n",
+        "  animeVae = True #@param {'type':'boolean'}\n",
+        "  SDVae = False #@param {'type':'boolean'}\n",
+        "  if animeVae:\n",
+        "    !wget -c https://huggingface.co/acheong08/secretAI/resolve/main/stableckpt/animevae.pt -O {root_dir}/stable-diffusion-webui/models/Stable-diffusion/novelAI.vae.pt\n",
+        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-webui/models/Stable-diffusion/novelAI.vae.pt\"\n",
+        "  if SDVae:\n",
+        "    !wget -c https://huggingface.co/stabilityai/sd-vae-ft-mse/resolve/main/diffusion_pytorch_model.bin -O {root_dir}/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5.vae.pt\n",
+        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5.vae.pt\"\n",
+        "\n",
         "  %cd {root_dir}/stable-diffusion-webui/\n",
         "  vram = \"--medvram\" #@param [\"--medvram\", \"--lowvram\", \"\"]\n",
         "  other_args = \"--share\" #@param [\"--share\", \"\"]\n",