Save images to google drive only

Author: Antonio Cheong

Diffusion_WebUI.ipynb

@@ -113,7 +113,7 @@
         "#@markdown SDv2.0 and mdjrny-v4 style has been added\n",
         "defaultModelURLs = [\"\" ,\"https://huggingface.co/acheong08/secretAI/resolve/main/stableckpt/animefull-final-pruned/model.ckpt\", \"https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt\", \"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt\", \"https://huggingface.co/Deltaadams/HentaiDiffusion/resolve/main/HD-16.ckpt\", \"https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt\", \"https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt\", \"https://huggingface.co/acheong08/f222/resolve/main/f222.ckpt\", \"https://huggingface.co/acheong08/Anything/resolve/main/Anything-V3.0-pruned-fp32.ckpt\", \"https://huggingface.co/acheong08/r34/resolve/main/r34_e4.ckpt\", \"https://huggingface.co/acheong08/HassansBlend/resolve/main/HassansBlend.ckpt\", \"https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt\", \"https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.ckpt\"]\n",
         "defaultModels = [\"\" ,\"NovelAI\", \"SD1-4\", \"WaifuDiffusion\", \"HentaiDiffusion\", \"SD1-5\", \"SD-inpainting\", \"f222-NSFW\", \"Anything-V3\", \"Rule34\", \"HassansBlend\", \"SDv2-0\", \"OpenMidjourney\"]\n",
-        "defaultModel = \"\" #@param [\"\" ,\"NovelAI\", \"SD1-4\", \"WaifuDiffusion\", \"HentaiDiffusion\", \"SD1-5\", \"SD-inpainting\", \"f222-NSFW\", \"Anything-V3\", \"Rule34\", \"HassansBlend\", \"SDv2-0\", \"OpenMidjourney\"]\n",
+        "defaultModel = \"Anything-V3\" #@param [\"\" ,\"NovelAI\", \"SD1-4\", \"WaifuDiffusion\", \"HentaiDiffusion\", \"SD1-5\", \"SD-inpainting\", \"f222-NSFW\", \"Anything-V3\", \"Rule34\", \"HassansBlend\", \"SDv2-0\", \"OpenMidjourney\"]\n",
         "\n",
         "#@markdown # Parameters\n",
         "#@markdown Select this if the model is based on SD2.0\n",
@@ -194,10 +194,11 @@
         "\n",
         "if run_in_gdrive:\n",
         "  drive.mount('/content/drive')\n",
-        "  !mkdir -p {'/content/drive/MyDrive/AI'}\n",
-        "  root_dir = \"/content/drive/MyDrive/AI\"\n",
+        "  !mkdir -p {'/content/drive/MyDrive/AI/output'}\n",
+        "  output_path = \"--img-output-dir /content/drive/MyDrive/AI/output/\"\n",
         "else:\n",
         "  root_dir = \"/content\"\n",
+        "  output_path = \"\"\n",
         "\n",
         "%cd {root_dir}\n",
         "\n",
@@ -245,6 +246,7 @@
         "  !git clone https://github.com/acheong08/stable-diffusion-webui\n",
         "  %cd {root_dir}/stable-diffusion-webui/\n",
         "  !git pull\n",
+        "  !git checkout dev\n",
         "  %cd {root_dir}\n",
         "  use_hypernetworks = False #@param {'type':'boolean'}\n",
         "\n",
@@ -281,7 +283,7 @@
         "    %cd /content/stable-diffusion-webui\n",
         "    !sed -i \"s/'cpu'/devices.get_optimal_device()/g\" modules/extras.py\n",
         "    vram = \"--lowvram\"\n",
-        "  !COMMANDLINE_ARGS=\"{other_args} {vae_args} {vram} --gradio-auth {gradio_username}:{gradio_password}\" REQS_FILE=\"requirements.txt\" python launch.py\n",
+        "  !COMMANDLINE_ARGS=\"{other_args} {vae_args} {vram} --gradio-auth {gradio_username}:{gradio_password} {output_path}\" REQS_FILE=\"requirements.txt\" python launch.py\n",
         "\n",
         "install_deps()\n",
         "run_webui()\n",
@@ -367,4 +369,4 @@
       "outputs": []
     }
   ]
-}
+}