Diffusion_UI.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "collapsed_sections": [
        "_xPLSbzNTgF2"
      ],
      "private_outputs": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU",
    "gpuClass": "standard"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "source": [
        "# Diffusion WebUI\n",
        "Includes:\n",
        "- Anything 3.0, 4.0, 4.5\n",
        "- NovelAI\n",
        "- Stable Diffusion (V2.1, V2, V1.5 and V1.4)\n",
        "- Waifu Diffusion\n",
        "- And many more\n",
        "\n",
        "Gradio not working correctly? To fix this use ngrok. Register if you don't have an account and change other-args to `--ngrok <token>`"
      ],
      "metadata": {
        "id": "QOmvhMFXLjbI"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "### [GitHub](https://github.com/misobarisic/Diffusion-UI)\n",
        "# Quick start: \n",
        "Run the Unified code.\n",
        "\n",
        "(If there are any issues, report them on [GitHub](https://github.com/misobarisic/Diffusion-UI)"
      ],
      "metadata": {
        "id": "lOC4X1umLtAi"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Check dependencies (Run this if you get errors)\n",
        "import os\n",
        "\n",
        "!nvidia-smi\n",
        "!pip install pytorch_lightning\n",
        "import pytorch_lightning as pl\n",
        "print(pl.__version__)\n"
      ],
      "metadata": {
        "id": "twBYi6Y4aj_a"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "# Unified code"
      ],
      "metadata": {
        "id": "TL9o4lz5XK2y"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Run this\n",
        "#@markdown # Instructions:\n",
        "#@markdown Select your favorite model (Or all of them)\n",
        "#@markdown 1. Run this cell.\n",
        "#@markdown 2. Wait\n",
        "#@markdown 3. Open gradio link\n",
        "\n",
        "#@markdown #### The default username is `webui` and password is `diffusion`\n",
        "\n",
        "#@markdown # Options\n",
        "import os\n",
        "import time\n",
        "from subprocess import getoutput\n",
        "import base64\n",
        "\n",
        "!pip3 install triton\n",
        "!pip install pytorch_lightning\n",
        "vae_args = \"\"\n",
        "large_models = False\n",
        "run_in_gdrive = False #@param {'type':'boolean'}\n",
        "optimization = \"None\" #@param [\"--opt-sdp-attention\", \"--opt-sdp-no-mem-attention\", \"None\"]\n",
        "\n",
        "#@markdown #### if your webui crashes when loading models, generating images or merging models, consider lowering the vram option\n",
        "vram = \"--medvram\" #@param [\"--highvram\", \"--medvram\", \"--lowvram\", \"--lowvram (large models)\"]\n",
        "if vram == \"--highvram\":\n",
        "  print(\"--highvram is not a real option. It is just a sentinel value to mimic not using --medvram or --lowvram.\")\n",
        "  vram = \"\"\n",
        "elif vram == \"--lowvram (large models)\":\n",
        "    vram = \"--lowvram\"\n",
        "    large_models = True\n",
        "\n",
        "other_args = \"--share --gradio-queue\" #@param [\"--share\", \"\"] {allow-input: true}\n",
        "gradio_username = \"webui\" #@param {'type': 'string'}\n",
        "gradio_password = \"diffusion\" #@param {'type': 'string'}\n",
        "#@markdown #### specifying your own Hugging Face user token allows you to download models from your own private repositories\n",
        "hf_user_token = \"\" #@param {'type': 'string'}\n",
        "#@markdown Choose the extensions you want\n",
        "tag_complete = False #@param {'type': 'boolean'}\n",
        "use_lora = False #@param {'type': 'boolean'}\n",
        "\n",
        "root_dir = \"/content\"\n",
        "\n",
        "if run_in_gdrive:\n",
        "  from google.colab import drive\n",
        "  drive.mount('/content/drive')\n",
        "  !mkdir -p {'/content/drive/MyDrive/AI'}\n",
        "  root_dir = \"/content/drive/MyDrive/AI\"\n",
        "\n",
        "\n",
        "blasphemy=base64.b64decode(\"d2VidWk=\").decode('ascii')\n",
        "\n",
        "%cd {root_dir}\n",
        "\n",
        "def get_hypernetworks():\n",
        "  !mkdir -p {root_dir}/stable-diffusion-$blasphemy/models/hypernetworks\n",
        "  hypernetworks = ['anime_2.pt', 'anime.pt', 'anime_3.pt', 'furry_2.pt', 'furry_3.pt', 'furry_protogen.pt', 'furry_transformation.pt', 'furry_scalie.pt', 'pony.pt', 'aini.pt', 'furry.pt', 'furry_kemono.pt']\n",
        "  for network in hypernetworks:\n",
        "    !wget -c https://huggingface.co/acheong08/secretAI/resolve/main/stableckpt/modules/modules/{network} -O {root_dir}/stable-diffusion-webui/models/hypernetworks/{network}\n",
        "\n",
        "def custom_model(url, checkpoint_name, format=0):\n",
        "  user_token = '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'\n",
        "  if hf_user_token != \"\":\n",
        "    user_token = hf_user_token\n",
        "  user_header = f\"\\\"Authorization: Bearer {user_token}\\\"\"\n",
        "  ext = \"\"\n",
        "  if format == 0:\n",
        "    ext = \"ckpt\"\n",
        "  elif format == 1:\n",
        "    ext = \"safetensors\"\n",
        "\n",
        "  !wget -c --header={user_header} {url} -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/{checkpoint_name}.{ext}\n",
        "\n",
        "def custom_lora(url, checkpoint_name, format=0):\n",
        "  user_token = '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'\n",
        "  if hf_user_token != \"\":\n",
        "    user_token = hf_user_token\n",
        "  user_header = f\"\\\"Authorization: Bearer {user_token}\\\"\"\n",
        "  ext = \"\"\n",
        "  if format == 0:\n",
        "    ext = \"ckpt\"\n",
        "  elif format == 1:\n",
        "    ext = \"safetensors\"\n",
        "\n",
        "  !wget -c --header={user_header} {url} -O {root_dir}/stable-diffusion-$blasphemy/models/Lora/{checkpoint_name}.{ext}\n",
        "\n",
        "\n",
        "def custom_embed(url, embed_name):\n",
        "  user_token = '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'\n",
        "  if hf_user_token != \"\":\n",
        "    user_token = hf_user_token\n",
        "  user_header = f\"\\\"Authorization: Bearer {user_token}\\\"\"\n",
        "  !wget -c --header={user_header} {url} -O {root_dir}/stable-diffusion-$blasphemy/embeddings/{embed_name}.pt\n",
        "\n",
        "\n",
        "def get_config(url, name):\n",
        "  !wget -c {url} -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/{name}.yaml\n",
        "\n",
        "\n",
        "def install_deps():\n",
        "  !pip install --upgrade fastapi==0.90.1\n",
        "\n",
        "  %cd {root_dir}\n",
        "  !git clone https://github.com/AUTOMATIC1111/stable-diffusion-$blasphemy\n",
        "  if tag_complete:\n",
        "    !git clone https://github.com/DominikDoom/a1111-sd-webui-tagcomplete /content/stable-diffusion-$blasphemy/extensions/a1111-sd-webui-tagcomplete\n",
        "  if use_lora:\n",
        "    !git clone https://github.com/kohya-ss/sd-webui-additional-networks.git /content/stable-diffusion-$blasphemy/extensions/sd-webui-additional-networks\n",
        "\n",
        "  #@markdown Choose the models you want\n",
        "  use_hypernetworks = False #@param {'type':'boolean'}\n",
        "  Anything_3 = False #@param {'type':'boolean'}\n",
        "  #Anything3Berrymix = False #@param {'type':'boolean'}\n",
        "  Anything_4_0 = False #@param {'type':'boolean'}\n",
        "  Anything_4_5 = False #@param {'type':'boolean'}\n",
        "  NovelAI = False #@param {'type':'boolean'}\n",
        "  Waifu_Diffusion_1_5_beta1 = False #@param {'type':'boolean'}\n",
        "  Waifu_Diffusion = False #@param {'type':'boolean'}\n",
        "  Hentai_Diffusion = False #@param {'type':'boolean'}\n",
        "  Gyokai = False #@param {'type':'boolean'}\n",
        "  Elysium_Anime_v3 = False #@param {'type':'boolean'}\n",
        "  Elysium_Anime_v2 = False #@param {'type':'boolean'}\n",
        "  Wlopmix = False #@param {'type':'boolean'}\n",
        "  Eimis_Anime_Diffusion = False #@param {'type':'boolean'}\n",
        "  Eimis_Semi_Realistic = False #@param {'type':'boolean'}\n",
        "  AbyssOrangeMix2_hard = False #@param {'type':'boolean'}\n",
        "  AbyssOrangeMix2_nsfw = False #@param {'type':'boolean'}\n",
        "  AbyssOrangeMix2_sfw = False #@param {'type':'boolean'}\n",
        "  AbyssOrangeMix3 = False #@param {'type':'boolean'}\n",
        "  #Kenshi = False #@param {'type':'boolean'}\n",
        "  PastelMix = False #@param {'type':'boolean'}\n",
        "  AnyPastel = False #@param {'type':'boolean'}\n",
        "  Counterfeit_2_5 = False #@param {'type': 'boolean'}\n",
        "  Rev_Animated = False #@param {'type': 'boolean'}\n",
        "  Deliberate_2 = False #@param {'type': 'boolean'}\n",
        "  CetusMix = False #@param {'type': 'boolean'}\n",
        "  Meina_7 = False #@param {'type': 'boolean'}\n",
        "  Meina_8 = False #@param {'type': 'boolean'}\n",
        "  Meina_9 = False #@param {'type': 'boolean'}\n",
        "  Meina_Hentai = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2_1 = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2_1_base = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2 = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2_base = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2_depth = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_2_inpainting = False #@param {'type': 'boolean'}\n",
        "  Stable_Diffusion_1_5 = False #@param {'type':'boolean'}\n",
        "  Stable_Diffusion_1_5_inpainting = False #@param {'type':'boolean'}\n",
        "  Stable_Diffusion = False #@param {'type':'boolean'}\n",
        "  Protogen_2_2 = False #@param {'type':'boolean'}\n",
        "  Protogen_2_2_pruned = False #@param {'type':'boolean'}\n",
        "  Protogen_3_4 = False #@param {'type':'boolean'}\n",
        "  Protogen_3_4_pruned = False #@param {'type':'boolean'}\n",
        "  Protogen_5_3 = False #@param {'type':'boolean'}\n",
        "  Protogen_5_3_pruned = False #@param {'type':'boolean'}\n",
        "  Protogen_5_8 = False #@param {'type':'boolean'}\n",
        "  Protogen_5_8_pruned = False #@param {'type':'boolean'}\n",
        "  Realistic_Vision_1_4 = False #@param {'type':'boolean'}\n",
        "  Realistic_Vision_2_0 = False #@param {'type':'boolean'}\n",
        "  Dreamshaper_4 = False #@param {'type':'boolean'}\n",
        "  Midjourney_v4_Diffusion = False #@param {'type':'boolean'}\n",
        "  HassanBlend = False #@param {'type':'boolean'}\n",
        "  F222 = False #@param {'type':'boolean'}\n",
        "\n",
        "  if Anything_3:\n",
        "    custom_model(\"https://huggingface.co/admruul/anything-v3.0/resolve/main/Anything-V3.0-pruned.safetensors\", \"anything3pruned\", 1)\n",
        "  #if Anything3Berry:\n",
        "  #  custom_model(\"https://huggingface.co/misobarisic/anything-berrymix/resolve/main/anything70_berry30.ckpt\", \"anything70_berry30\")\n",
        "  if Anything_4_0:\n",
        "    custom_model(\"https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0-pruned.safetensors\", \"anything-v4.0\", 1)\n",
        "  if Anything_4_5:\n",
        "    custom_model(\"https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors\", \"anything-v4.5\", 1)\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",
        "    custom_model(\"https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt\", \"sd-v1.4\")\n",
        "  if Waifu_Diffusion_1_5_beta1:\n",
        "    custom_model(\"https://huggingface.co/waifu-diffusion/wd-1-5-beta/resolve/main/checkpoints/wd15-beta1-fp32.safetensors\", \"wd15-beta-1\", 1)\n",
        "    get_config(\"https://huggingface.co/waifu-diffusion/wd-1-5-beta/raw/main/checkpoints/wd15-beta1-fp32.yaml\", \"wd15-beta-1\")\n",
        "  if Waifu_Diffusion:\n",
        "    custom_model(\"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt\", \"waifu_diffusion\")\n",
        "  if Hentai_Diffusion:\n",
        "    custom_model(\"https://huggingface.co/Deltaadams/HentaiDiffusion/resolve/main/HD-17.ckpt\", \"h_diffusion_17\")\n",
        "  if Gyokai:\n",
        "    custom_model(\"https://huggingface.co/NoCrypt/gyokai/resolve/main/gyokai.ckpt\", \"gyokai\")\n",
        "  if Elysium_Anime_v3:\n",
        "    custom_model(\"https://huggingface.co/hesw23168/SD-Elysium-Model/resolve/main/Elysium_Anime_V3.safetensors\", \"elysium_anime_v3\", 1)\n",
        "  if Elysium_Anime_v2:\n",
        "    custom_model(\"https://huggingface.co/hesw23168/SD-Elysium-Model/resolve/main/Elysium_Anime_V2.ckpt\", \"elysium_anime_v2\")\n",
        "  if Wlopmix:\n",
        "    custom_model(\"https://huggingface.co/SirVeggie/wlop/resolve/main/wlopmix.ckpt\", \"wlopmix\")\n",
        "  if Eimis_Anime_Diffusion:\n",
        "    custom_model(\"https://huggingface.co/eimiss/EimisAnimeDiffusion_1.0v/resolve/main/EimisAnimeDiffusion_1-0v.ckpt\", \"eimis_anime_diffusion_v1.0\")\n",
        "  if Eimis_Semi_Realistic:\n",
        "    custom_model(\"https://huggingface.co/eimiss/EimisSemiRealistic/resolve/main/EimisSemiRealistic_1-0v.ckpt\", \"eimis_semi_realistic_v1.0\")\n",
        "  if Stable_Diffusion_2_1:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-nonema-pruned.ckpt\", \"sd-2-1-768\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml\", \"sd-2-1-768\")\n",
        "  if Stable_Diffusion_2_1_base:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-nonema-pruned.ckpt\", \"sd-2-1-512\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml\", \"sd-2-1-512\")\n",
        "  if Stable_Diffusion_2:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt\", \"sd-2-768\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml\", \"sd-2-768\")\n",
        "  if Stable_Diffusion_2_base:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt\", \"sd-2-512\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml\", \"sd-2-512\")\n",
        "  if Stable_Diffusion_2_depth:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2-depth/resolve/main/512-depth-ema.ckpt\", \"sd-2-depth\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml\", \"sd-2-depth\")\n",
        "  if Stable_Diffusion_2_inpainting:\n",
        "    custom_model(\"https://huggingface.co/stabilityai/stable-diffusion-2-inpainting/resolve/main/512-inpainting-ema.ckpt\", \"sd-2-inpainting\")\n",
        "    get_config(\"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inpainting-inference-v.yaml\", \"sd-2-inpainting\")\n",
        "  if Stable_Diffusion_1_5:\n",
        "    custom_model(\"https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt\", \"sd-v1-5\")\n",
        "  if Stable_Diffusion_1_5_inpainting:\n",
        "    custom_model(\"https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt\", \"sd-v1-5-inpainting\")\n",
        "  if HassanBlend:\n",
        "    custom_model(\"https://huggingface.co/hassanblend/hassanblend1.4/resolve/main/HassanBlend1.4_Safe.safetensors\", \"hassanblend1.4\", 1)\n",
        "  if F222:\n",
        "    custom_model(\"https://huggingface.co/acheong08/f222/resolve/main/f222.ckpt\", \"f222\")\n",
        "  if Midjourney_v4_Diffusion:\n",
        "    custom_model(\"https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.safetensors\", \"midjourney-v4-diffusion\", 1)\n",
        "  if AbyssOrangeMix2_hard:\n",
        "    custom_model(\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_hard.safetensors\", \"AbyssOrangeMix2_hard\", 1)\n",
        "  if AbyssOrangeMix2_nsfw:\n",
        "    custom_model(\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_nsfw.safetensors\", \"AbbysOrangeMix2_nsfw\", 1)\n",
        "  if AbyssOrangeMix2_sfw:\n",
        "    custom_model(\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_sfw.safetensors\", \"AbyssOrangeMix2_sfw\", 1)\n",
        "  if AbyssOrangeMix3:\n",
        "    custom_model(\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix3/AOM3_orangemixs.safetensors\", \"AOM3\", 1)\n",
        "  #if Kenshi:\n",
        "  #  custom_model(\"https://huggingface.co/SweetLuna/Kenshi/blob/main/KENSHI%2001/KENSHI01_Pruned.safetensors\", \"Kenshi01\", 1)\n",
        "  if PastelMix:\n",
        "    custom_model(\"https://huggingface.co/andite/pastel-mix/resolve/main/pastelmix-better-vae-fp32.safetensors\", \"pastelmix\", 1)\n",
        "  if AnyPastel:\n",
        "    custom_model(\"https://huggingface.co/m4gnett/any-pastel/resolve/main/AnyPastel-fp16.safetensors\", \"anypastel\", 1)\n",
        "  if Counterfeit_2_5:\n",
        "    custom_model(\"https://huggingface.co/gsdf/Counterfeit-V2.5/resolve/main/Counterfeit-V2.5_pruned.safetensors\",\"Counterfeit-V2.5\" , 1)\n",
        "  if Protogen_5_3:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x5.3_Official_Release/resolve/main/ProtoGen_X5.3.safetensors\", \"Protogen_X5.3\", 1)\n",
        "  if Protogen_5_3_pruned:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x5.3_Official_Release/resolve/main/ProtoGen_X5.3-pruned-fp16.safetensors\", \"Protogen_X5.3_pruned\", 1)\n",
        "  if Protogen_5_8:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x5.8_Official_Release/resolve/main/ProtoGen_X5.8.safetensors\", \"Protogen_X5.8\", 1)\n",
        "  if Protogen_5_8_pruned:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x5.8_Official_Release/resolve/main/ProtoGen_X5.8-pruned-fp16.safetensors\", \"Protogen_X5.8_pruned\", 1)\n",
        "  if Protogen_2_2:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_v2.2_Official_Release/resolve/main/Protogen_V2.2.safetensors\", \"Protogen_V2.2\", 1)\n",
        "  if Protogen_2_2_pruned:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_v2.2_Official_Release/resolve/main/Protogen_V2.2-pruned-fp16.safetensors\", \"Protogen_V2.2_pruned\", 1)\n",
        "  if Protogen_3_4:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/resolve/main/ProtoGen_X3.4.safetensors\", \"Protogen_X3.4\", 1)\n",
        "  if Protogen_3_4_pruned:\n",
        "    custom_model(\"https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/resolve/main/Protogen_X3.4-pruned-fp16.safetensors\", \"Protogen_X3.4_pruned\", 1)\n",
        "  if Meina_7:\n",
        "    custom_model(\"https://huggingface.co/Meina/MeinaMix/resolve/main/Meina%20V7%20-%20pruned.safetensors\", \"Meina_v7\", 1)\n",
        "  if Meina_8:\n",
        "    custom_model(\"https://huggingface.co/Meina/MeinaMix/resolve/main/Meina%20V8%20pruned.safetensors\", \"Meina_v8\", 1)\n",
        "  if Meina_9:\n",
        "    custom_model(\"https://huggingface.co/Meina/MeinaMix/resolve/main/Meina%20V9%20-%20baked%20VAE.safetensors\", \"Meina_v9\", 1)\n",
        "  if Meina_Hentai:\n",
        "    custom_model(\"https://huggingface.co/Meina/MeinaMix/resolve/main/MeinaHentai%20-%20baked%20VAE.safetensors\", \"Meina_Hentai\", 1)\n",
        "  if Realistic_Vision_1_4:\n",
        "    custom_model(\"https://huggingface.co/SG161222/Realistic_Vision_V1.4/resolve/main/Realistic_Vision_V1.4.safetensors\",\"Realistic_Vision_V1.4\", 1)\n",
        "  if Realistic_Vision_2_0:\n",
        "    custom_model(\"https://huggingface.co/SG161222/Realistic_Vision_V2.0/resolve/main/Realistic_Vision_V2.0.safetensors\",\"Realistic_Vision_V2.0\", 1)\n",
        "  if Dreamshaper_4:\n",
        "    custom_model(\"https://huggingface.co/Lykon/DreamShaper/resolve/main/DreamShaper_4BakedVae_fp16.safetensors\", \"DreamSHaper_4BakedVae\", 1)\n",
        "  if Rev_Animated:\n",
        "    custom_model(\"https://huggingface.co/ahmmu20/rev_animated/resolve/main/revAnimated_v11.safetensors\", \"revAnimated_v11\", 1)\n",
        "  if Deliberate_2:\n",
        "    custom_model(\"https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v2.safetensors\", \"Deliberate_v2\", 1)\n",
        "  if CetusMix:\n",
        "    custom_model(\"https://huggingface.co/ckpt/cetus-mix/resolve/main/cetusMix_Version35.safetensors\", \"cetusMis_Version35\", 1)\n",
        "\n",
        "  try:\n",
        "    for (name, url, config_url, format_type) in models:\n",
        "      custom_model(url, name, format_type)\n",
        "      if config_url != None:\n",
        "        get_config(config_url, name)\n",
        "  except:\n",
        "    pass\n",
        "\n",
        "  if use_hypernetworks:\n",
        "    get_hypernetworks()\n",
        "    \n",
        "  %cd {root_dir}/stable-diffusion-$blasphemy/extensions\n",
        "  !git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser\n",
        "  %cd {root_dir}\n",
        "\n",
        "\n",
        "def install_embeds():\n",
        "  try:\n",
        "    for (name, url) in embeds:\n",
        "      custom_embed(url, name)\n",
        "  except:\n",
        "    pass\n",
        "    \n",
        "def install_lora():\n",
        "  user_token = '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'\n",
        "  if hf_user_token != \"\":\n",
        "    user_token = hf_user_token\n",
        "  user_header = f\"\\\"Authorization: Bearer {user_token}\\\"\"\n",
        "  \n",
        "  #@markdown Choose the LoRA to include\n",
        "  raiden_shogun_LoRA = False #@param {'type':'boolean'}\n",
        "  yae_miko_Mediumpruned = False #@param {'type':'boolean'}\n",
        "  Shenhe_Medium = False #@param {'type':'boolean'}\n",
        "  eulaMedium = False #@param {'type':'boolean'}\n",
        "  Ganyu_Medium = False #@param {'type':'boolean'}\n",
        "  #@markdown Additional LORAs can be added by running the appropriate cell down below\n",
        "  \n",
        "  if raiden_shogun_LoRA:\n",
        "    !wget -P {root_dir}/stable-diffusion-$blasphemy/models/Lora https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Raiden%20Shogun/raiden%20shogun_LoRA.safetensors\n",
        "  if yae_miko_Mediumpruned:\n",
        "    !wget -P {root_dir}/stable-diffusion-$blasphemy/models/Lora https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/yae%20miko/yae%20miko_Mediumpruned.safetensors\n",
        "  if Shenhe_Medium:\n",
        "    !wget -P {root_dir}/stable-diffusion-$blasphemy/models/Lora https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Shenhe/Shenhe_Medium.safetensors\n",
        "  if eulaMedium:\n",
        "    !wget -P {root_dir}/stable-diffusion-$blasphemy/models/Lora https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Eula/eulaMedium.safetensors\n",
        "  if Ganyu_Medium:\n",
        "    !wget -P {root_dir}/stable-diffusion-$blasphemy/models/Lora https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Ganyu/Ganyu_Medium.safetensors\n",
        "\n",
        "  try:\n",
        "    for (name, url, num) in lora:\n",
        "      custom_lora(url, name, num)\n",
        "  except:\n",
        "    pass\n",
        "\n",
        "\n",
        "def run_webui():\n",
        "  #@markdown Choose the vae you want\n",
        "  vae = \"Anime (Anything 4)\" #@param [\"Anime (Anything 3)\", \"Anime (Anything 4)\", \"Anime (Orangemix)\",\"Anime (Waifu Diffusion 1.4)\", \"Stable Diffusion\", \"None\"]\n",
        "  \n",
        "  if vae == \"Anime (Anything 3)\":\n",
        "    !wget -c https://huggingface.co/admruul/anything-v3.0/resolve/main/Anything-V3.0.vae.pt -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/Anything-V3.0.vae.pt\n",
        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-\" + blasphemy + \"/models/Stable-diffusion/Anything-V3.0.vae.pt\"\n",
        "  if vae == \"Anime (Anything 4)\":\n",
        "    !wget -c https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/anything-v4.0.vae.pt\n",
        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-\" + blasphemy + \"/models/Stable-diffusion/anything-v4.0.vae.pt\"\n",
        "  if vae == \"Anime (Orangemix)\":\n",
        "    !wget -c https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/orangemix.vae.pt\n",
        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-\" + blasphemy + \"/models/Stable-diffusion/orangemix.vae.pt\"\n",
        "  if vae == \"Anime (Waifu Diffusion 1.4)\":\n",
        "    !wget -c https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime.ckpt -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/kl-f8-anime.vae.pt\n",
        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-\" + blasphemy + \"/models/Stable-diffusion/kl-f8-anime.vae.pt\"\n",
        "  if vae == \"Stable Diffusion\":\n",
        "    !wget https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt -O {root_dir}/stable-diffusion-$blasphemy/models/Stable-diffusion/sd-v1-5.vae.pt\n",
        "    vae_args = \"--vae-path \" + root_dir + \"/stable-diffusion-\" + blasphemy + \"/models/Stable-diffusion/sd-v1-5.vae.pt\"\n",
        "\n",
        "  %cd {root_dir}/stable-diffusion-$blasphemy/\n",
        "  !COMMANDLINE_ARGS=\"{other_args} {vae_args} {vram} --gradio-auth {gradio_username}:{gradio_password}\" REQS_FILE=\"requirements.txt\" python launch.py\n",
        "\n",
        "install_deps()\n",
        "install_embeds()\n",
        "install_lora()\n",
        "if optimization != \"None\":\n",
        "  other_args = f\"{optimization} {other_args}\"\n",
        "\n",
        "if large_models:\n",
        "  %cd /content/stable-diffusion-$blasphemy\n",
        "  !sed -i \"s/'cpu'/devices.get_optimal_device()/g\" modules/extras.py\n",
        "\n",
        "run_webui()"
      ],
      "metadata": {
        "id": "Y4qjIc1XXKWw"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Additional models\n",
        "\n",
        "Run the cell for each model you want to download.\n",
        "Model will be downloaded once the cell above is ran."
      ],
      "metadata": {
        "id": "gk0VPCRn03mA"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "model_name = \"AnyWolpGyokaiElysium\" #@param {'type': 'string'}\n",
        "model_url = \"https://huggingface.co/misobarisic/mixes/resolve/main/AnyWolpGyokaiElysium.ckpt\" #@param {'type': 'string'}\n",
        "format = \"autodetect\" #@param [\"autodetect\", \"ckpt\", \"safetensors\"]\n",
        "\n",
        "try:\n",
        "  models\n",
        "except:\n",
        "  models = set()\n",
        "\n",
        "#@markdown If you need a custom `.yaml` file for a model. Set it here, else leave the field blank.\n",
        "config_url = \"\" #@param {'type': 'string'}\n",
        "if config_url == \"\":\n",
        "  config_url = None\n",
        "\n",
        "model_num = None\n",
        "if format == \"autodetect\":\n",
        "  if model_url.endswith(\".ckpt\"):\n",
        "    model_num = 0\n",
        "  elif model_url.endswith(\".safetensors\"):\n",
        "    model_num = 1\n",
        "  else:\n",
        "    raise Exception(\"Could not autodetect weight save format\")\n",
        "elif format == \"ckpt\":\n",
        "  model_num = 0\n",
        "elif format == \"safetensors\":\n",
        "  model_num = 1\n",
        "\n",
        "models.add((model_name, model_url, config_url, model_num))\n"
      ],
      "metadata": {
        "id": "fuigrT-V-acV"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Clear model set"
      ],
      "metadata": {
        "id": "hCOwvZTQxsTq"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "models = set()"
      ],
      "metadata": {
        "id": "sEpVo7tQx0vE"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Select embeddings\n",
        "\n",
        "Run the cells for each embedding you want to download.\n",
        "Embeddings will be downloaded once the cell above is ran."
      ],
      "metadata": {
        "id": "3scgdaso-a5_"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "embed = \"bad_prompt\" #@param [\"bad_prompt\"]\n",
        "\n",
        "try:\n",
        "  embeds\n",
        "except:\n",
        "  embeds = set()\n",
        "\n",
        "embed_url = \"\"\n",
        "if embed == \"bad_prompt\":\n",
        "  embed_url = \"https://huggingface.co/datasets/Nerfgun3/bad_prompt/resolve/main/bad_prompt_version2.pt\"\n",
        " \n",
        "embeds.add((embed, embed_url))"
      ],
      "metadata": {
        "id": "JdbD-Mmo0lRx"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "embed_name = \"bad_prompt\" #@param {'type': 'string'}\n",
        "embed_url = \"https://huggingface.co/datasets/Nerfgun3/bad_prompt/resolve/main/bad_prompt_version2.pt\" #@param {'type': 'string'}\n",
        "\n",
        "try:\n",
        "  embeds\n",
        "except:\n",
        "  embeds = set()\n",
        "\n",
        "embeds.add((embed, embed_url))"
      ],
      "metadata": {
        "id": "VNouRPey3lzg"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Clear embeddings set"
      ],
      "metadata": {
        "id": "QUBn_Ta8yDIG"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "embeds = set()"
      ],
      "metadata": {
        "id": "rJliIWe-yDIH"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Additional LoRAs\n",
        "\n",
        "Run the cells for each LoRA you want to download."
      ],
      "metadata": {
        "id": "VdmdX1Hipy7H"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "lora_name = \"RaidenShogun\" #@param {'type': 'string'}\n",
        "lora_url = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Raiden%20Shogun/raiden%20shogun_LoRA.safetensors\" #@param {'type': 'string'}\n",
        "format = \"autodetect\" #@param [\"autodetect\", \"ckpt\", \"safetensors\"]\n",
        "\n",
        "try:\n",
        "  lora\n",
        "except:\n",
        "  lora = set()\n",
        "\n",
        "model_num = None\n",
        "if format == \"autodetect\":\n",
        "  if lora_url.endswith(\".ckpt\"):\n",
        "    num = 0\n",
        "  elif lora_url.endswith(\".safetensors\"):\n",
        "    num = 1\n",
        "  else:\n",
        "    raise Exception(\"Could not autodetect weight save format\")\n",
        "elif format == \"ckpt\":\n",
        "  num = 0\n",
        "elif format == \"safetensors\":\n",
        "  num = 1\n",
        "\n",
        "lora.add((lora_name, lora_url, num))\n"
      ],
      "metadata": {
        "id": "xxtoTmkYp38F"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Clear LoRA set"
      ],
      "metadata": {
        "id": "fCLUNNMlp-qN"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "lora = set()"
      ],
      "metadata": {
        "id": "ikWRVY4yp-qO"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Saving your images"
      ],
      "metadata": {
        "id": "_xPLSbzNTgF2"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "#@markdown Download file manually from files tab or save to Google Drive\n",
        "%cd /content/stable-diffusion-webui/\n",
        "!zip -r /content/output.zip outputs\n",
        "\n",
        "from pydrive.auth import GoogleAuth\n",
        "from google.colab import drive\n",
        "from pydrive.drive import GoogleDrive\n",
        "from google.colab import auth\n",
        "from oauth2client.client import GoogleCredentials\n",
        "def create_folder(folder_name):\n",
        "    # Check if folder exists\n",
        "    file_list = drive.ListFile({'q': \"title='{}' and mimeType='application/vnd.google-apps.folder' and trashed=false\".format(folder_name)}).GetList()\n",
        "    if len(file_list) > 0:\n",
        "        # Folder exists\n",
        "        print('Debug: Folder exists')\n",
        "        folder_id = file_list[0]['id']\n",
        "    else:\n",
        "        print('Debug: Creating folder')\n",
        "        file = drive.CreateFile({'title': folder_name, 'mimeType': 'application/vnd.google-apps.folder'})\n",
        "        file.Upload()\n",
        "        folder_id = file.attr['metadata']['id']\n",
        "    # return folder id\n",
        "    return folder_id\n",
        "# Upload file to Google Drive\n",
        "def upload_file(file_name, folder_id, save_as):\n",
        "    # Check if file exists\n",
        "    file_list = drive.ListFile({'q': \"title='{}' and trashed=false\".format(save_as)}).GetList()\n",
        "    if len(file_list) > 0:\n",
        "        print('Debug: File already exists')\n",
        "        # Change file name to avoid overwriting\n",
        "        save_as = save_as + ' (1)'\n",
        "    file = drive.CreateFile({'title': save_as, 'parents': [{'id': folder_id}]})\n",
        "    file.SetContentFile(file_name)\n",
        "    # Upload and set permission to public\n",
        "    file.Upload()\n",
        "    file.InsertPermission({'type': 'anyone', 'value': 'anyone', 'role': 'reader'})\n",
        "    # return file id\n",
        "    return file.attr['metadata']['id']\n",
        "\n",
        "use_drive = False #@param {type:\"boolean\"}\n",
        "folder_name = \"AI_pic_archive\" #@param {type: \"string\"}\n",
        "save_as = \"oni.zip\" #@param {type: \"string\"}\n",
        "\n",
        "if use_drive:\n",
        "  auth.authenticate_user()\n",
        "  gauth = GoogleAuth()\n",
        "  gauth.credentials = GoogleCredentials.get_application_default()\n",
        "  drive = GoogleDrive(gauth)\n",
        "  file_id = upload_file('/content/output.zip', create_folder(folder_name), save_as)\n",
        "  print(\"Your sharing link: https://drive.google.com/file/d/\" + file_id + \"/view?usp=sharing\")  "
      ],
      "metadata": {
        "id": "lpiP4AkYTic0"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}