Nios_test_.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/AI-Nios/stable-diffusion-webui-colab-japanese/blob/main/Nios_test_.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "3rvLOgqyDUdD",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title 1.WebUIのインストール\n",
        "%cd /content\n",
        "import os\n",
        "from google.colab.output import eval_js\n",
        "os.environ['colab_url'] = eval_js(\"google.colab.kernel.proxyPort(7860, {'cache': false})\")\n",
        "\n",
        "!apt -y update -qq\n",
        "!wget http://launchpadlibrarian.net/367274644/libgoogle-perftools-dev_2.5-2.2ubuntu3_amd64.deb\n",
        "!wget https://launchpad.net/ubuntu/+source/google-perftools/2.5-2.2ubuntu3/+build/14795286/+files/google-perftools_2.5-2.2ubuntu3_all.deb\n",
        "!wget https://launchpad.net/ubuntu/+source/google-perftools/2.5-2.2ubuntu3/+build/14795286/+files/libtcmalloc-minimal4_2.5-2.2ubuntu3_amd64.deb\n",
        "!wget https://launchpad.net/ubuntu/+source/google-perftools/2.5-2.2ubuntu3/+build/14795286/+files/libgoogle-perftools4_2.5-2.2ubuntu3_amd64.deb\n",
        "!apt install -qq libunwind8-dev\n",
        "!dpkg -i *.deb\n",
        "%env LD_PRELOAD=libtcmalloc.so\n",
        "!rm *.deb\n",
        "\n",
        "!apt -y install -qq aria2\n",
        "!pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 torchtext==0.14.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu116 -U\n",
        "!pip install -q xformers==0.0.16 triton==2.0.0 -U\n",
        "\n",
        "#WebUIのインストール\n",
        "!git clone -b v2.0 https://github.com/camenduru/stable-diffusion-webui\n",
        "!git clone https://github.com/alemelis/sd-webui-ar.git /content/stable-diffusion-webui/extensions/sd-webui-ar\n",
        "!git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /content/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser\n",
        "!git clone https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git /content/stable-diffusion-webui/extensions/a1111-sd-webui-tagcomplete\n",
        "!git clone https://github.com/camenduru/stable-diffusion-webui-huggingface /content/stable-diffusion-webui/extensions/stable-diffusion-webui-huggingface\n",
        "!git clone https://github.com/camenduru/sd-webui-tunnels /content/stable-diffusion-webui/extensions/sd-webui-tunnels\n",
        "\n",
        "#Extensions\n",
        "Extensions = True #@param {'type':'boolean'}\n",
        "if Extensions == True :\n",
        "  !git clone https://github.com/deforum-art/deforum-for-automatic1111-webui /content/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui\n",
        "  !git clone https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git /content/stable-diffusion-webui/extensions/dataset-tag-editor\n",
        "  !git clone https://github.com/bbc-mc/sdweb-merge-block-weighted-gui /content/stable-diffusion-webui/extensions/sdweb-merge-block-weighted-gui\n",
        "  !git clone -b v2.0 https://github.com/camenduru/sd-civitai-browser /content/stable-diffusion-webui/extensions/sd-civitai-browser\n",
        "  !git clone https://github.com/kohya-ss/sd-webui-additional-networks /content/stable-diffusion-webui/extensions/sd-webui-additional-networks\n",
        "  !git clone https://github.com/opparco/stable-diffusion-webui-two-shot.git /content/stable-diffusion-webui/extensions/stable-diffusion-webui-two-shot\n",
        "  !git apply --ignore-whitespace /extensions/stable-diffusion-webui-two-shot/cfg_denoised_callback-ea9bd9fc.patch\n",
        "  !git clone https://github.com/hallatore/stable-diffusion-webui-chatgpt-utilities /content/stable-diffusion-webui/extensions/stable-diffusion-webui-chatgpt-utilities\n",
        "  !git clone https://github.com/KohakuBlueleaf/a1111-sd-webui-locon.git /content/stable-diffusion-webui/extensions/a1111-sd-webui-locon\n",
        "\n",
        "%cd /content/stable-diffusion-webui\n",
        "!git reset --hard\n",
        "\n",
        "#ControlNet\n",
        "ControlNet = True #@param {'type':'boolean'}\n",
        "if ControlNet == True :\n",
        "  !git clone https://github.com/camenduru/openpose-editor /content/stable-diffusion-webui/extensions/openpose-editor\n",
        "  !git clone https://github.com/Mikubill/sd-webui-controlnet.git /content/stable-diffusion-webui/extensions/sd-webui-controlnet\n",
        "  !git clone https://github.com/jexom/sd-webui-depth-lib /content/stable-diffusion-webui/extensions/sd-webui-depth-lib\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_canny-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_canny-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_depth-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_depth-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_hed-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_hed-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_mlsd-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_mlsd-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_normal-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_normal-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_openpose-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_openpose-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_scribble-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_scribble-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/control_seg-fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_seg-fp16.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/hand_pose_model.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/openpose -o hand_pose_model.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/body_pose_model.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/openpose -o body_pose_model.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/dpt_hybrid-midas-501f0c75.pt -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/midas -o dpt_hybrid-midas-501f0c75.pt\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/mlsd_large_512_fp32.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/mlsd -o mlsd_large_512_fp32.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/mlsd_tiny_512_fp32.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/mlsd -o mlsd_tiny_512_fp32.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/network-bsds500.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/hed -o network-bsds500.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/upernet_global_small.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator/uniformer -o upernet_global_small.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_style_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_style_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_sketch_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_sketch_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_seg_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_seg_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_openpose_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_openpose_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_keypose_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_keypose_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_depth_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_depth_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_color_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_color_sd14v1.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet/resolve/main/t2iadapter_canny_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_canny_sd14v1.pth\n",
        "\n",
        "#EasyNegative\n",
        "!wget https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.pt -O /content/stable-diffusion-webui/embeddings/EasyNegative.pt\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "r8FuOir_Dkox",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title  2.モデル選択\n",
        "\n",
        "model_name =\"7th_anime_v3_A\" #@param [\"stable_diffusion_v1.5\",\"stable_diffusion_v2.1\",\"waifu_diffusion_v1.3\",\"Openjourney\",\"anything-v3.0\",\"anything-v4.0\",\"anything-v4.5\",\"Counterfeit-V2.0\",\"Counterfeit-V2.5\",\"8528-diffusion\",\"AbyssOrangeMix2\",\"EerieOrangeMix2\",\"trinart2_step115000\",\"Eimis Anime Diffusion_1.0v\",\"dreamlike-photoreal-2.0\",\"Cyberpunk-Anime-Diffusion\",\"Plat Diffusion v1.3.1\",\"ACertainThing\",\"pastel_mix\",\"Basil_mix_fixed\",\"Realistic Vision V1.3\",\"SukiyakiMix-v1.0\",\"7th_anime_v1.1\", \"7th_anime_v2_A\", \"7th_anime_v2_B\", \"7th_anime_v2_C\", \"7th_anime_v2_G\", \"7th_anime_v3_A\", \"7th_anime_v3_B\", \"7th_anime_v3_C\",\"7th_anime_3.1_B\",\"YuzuGinger_v4\",\"YuzuLemonMilk_v1.5\", \"Defmix-v1.1\",\"RDtMix\",\"meadmix\",\"Nabylon-v1.0\",\"LonganMix\",\"Corneo's 7th Heaven Mix\", \"Corneo's Shinra26 Mix\",\"atwmix_test2\",\"Graham_Special\", \"BalorV2featAbyssOrange2\",\"Balor-V2.5featAOM3A3\",\"ChilloutMix\", \"wd-1-5-beta2\"] {allow-input: false}\n",
        "if model_name == \"stable_diffusion_v1.5\":\n",
        "  url1  = \"https://huggingface.co/camenduru/sd15/resolve/main/v1-5-pruned-emaonly.ckpt\"\n",
        "  path1 = \"v1-5-pruned-emaonly.ckpt\"\n",
        "\n",
        "elif model_name == \"stable_diffusion_v2.1\":\n",
        "  url1  = \"https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.ckpt\" \n",
        "  path1 = \"v2-1_768-ema-pruned.ckpt\"\n",
        "\n",
        "elif model_name == \"waifu_diffusion_v1.3\":\n",
        "  url1  = \"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt\" \n",
        "  path1 = \"wd-v1-3-float32.ckpt\"\n",
        "\n",
        "elif model_name == \"Openjourney\":\n",
        "  url1  = \"https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.safetensors\"\n",
        "  path1 = \"mdjrny-v4.safetensors\"\n",
        "\n",
        "elif model_name == \"anything-v3.0\":\n",
        "  url1  = \"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/anything-v3-fp32-pruned.safetensors\"\n",
        "  path1 = \"anything-v3-fp32-pruned.safetensors\"\n",
        "  \n",
        "elif model_name == \"anything-v4.0\":\n",
        "  url1  = \"https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0-pruned.ckpt\"\n",
        "  path1 = \"anything-v4.0-pruned.ckpt\"\n",
        "\n",
        "elif model_name == \"anything-v4.5\":\n",
        "  url1  = \"https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors\" \n",
        "  path1 = \"anything-v4.5-pruned.safetensors\"\n",
        "\n",
        "elif model_name == \"Counterfeit-V2.0\":\n",
        "  url1  = \"https://huggingface.co/gsdf/Counterfeit-V2.0/resolve/main/Counterfeit-V2.0fp16.safetensors\"\n",
        "  path1 = \"Counterfeit-V2.0fp16.safetensors\"\n",
        "\n",
        "elif model_name == \"Counterfeit-V2.5\":\n",
        "  url1  = \"https://huggingface.co/gsdf/Counterfeit-V2.5/resolve/main/Counterfeit-V2.5_pruned.safetensors\"\n",
        "  path1 = \"Counterfeit-V2.5_pruned.safetensors\"\n",
        "\n",
        "elif model_name == \"8528-diffusion\":\n",
        "  url1  = \"https://huggingface.co/852wa/8528-diffusion/resolve/main/8528d-final.ckpt\"\n",
        "  path1 = \"8528d-final.ckpt\"\n",
        "\n",
        "elif model_name == \"AbyssOrangeMix2\":\n",
        "  url1  = \"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_nsfw.safetensors\" \n",
        "  path1 = \"AbyssOrangeMix2_sfw.safetensors\"\n",
        "\n",
        "elif model_name == \"EerieOrangeMix2\":\n",
        "  url1  = \"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/EerieOrangeMix/EerieOrangeMix2.safetensors \" \n",
        "  path1 = \"EerieOrangeMix2.safetensors\"\n",
        "\n",
        "elif model_name == \"trinart2_step115000\":\n",
        "  url1  = \"https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step115000.ckpt \" \n",
        "  path1 = \"trinart2_step115000.ckpt\"\n",
        "\n",
        "elif model_name == \"Eimis Anime Diffusion_1.0v\":\n",
        "  url1  = \"https://huggingface.co/eimiss/EimisAnimeDiffusion_1.0v/resolve/main/EimisAnimeDiffusion_1-0v.ckpt \" \n",
        "  path1 = \"EimisAnimeDiffusion_1-0v.ckpt\"\n",
        "\n",
        "elif model_name == \"dreamlike-photoreal-2.0\":\n",
        "  url1  = \"https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0/resolve/main/dreamlike-photoreal-2.0.safetensors\" \n",
        "  path1 = \"/content/stable-diffusion-webui/models/Stable-diffusion/dreamlike-photoreal-2.0.safetensors\"\n",
        "\n",
        "elif model_name == \"Cyberpunk-Anime-Diffusion\":\n",
        "  url1  = \"https://huggingface.co/DGSpitzer/Cyberpunk-Anime-Diffusion/resolve/main/Cyberpunk-Anime-Diffusion.ckpt\" \n",
        "  path1 = \"Cyberpunk-Anime-Diffusion.ckpt\"\n",
        "\n",
        "elif model_name == \"Plat Diffusion v1.3.1\":\n",
        "  url1  = \"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\" \n",
        "  path1 = \"plat-v1-3-1.safetensors\"\n",
        "\n",
        "elif model_name == \"ACertainThing\":\n",
        "  url1  = \"https://huggingface.co/JosephusCheung/ACertainThing/resolve/main/ACertainThing.ckpt\" \n",
        "  path1 = \"ACertainThing.ckpt\"\n",
        "\n",
        "elif model_name == \"pastel_mix\":\n",
        "  url1  = \"https://huggingface.co/andite/pastel-mix/resolve/main/pastelmix-better-vae-fp32.ckpt\" \n",
        "  path1 = \"pastel_mix.ckpt\"\n",
        "\n",
        "elif model_name == \"Basil_mix_fixed\":\n",
        "  url1  = \"https://huggingface.co/nuigurumi/basil_mix/resolve/main/Basil_mix_fixed.safetensors\" \n",
        "  path1 = \"Basil_mix_fixed.safetensors\"\n",
        "\n",
        "elif model_name == \"Realistic Vision V1.3\":\n",
        "  url1  = \"https://civitai.com/api/download/models/6987\" \n",
        "  path1 = \"Realistic_Vision.safetensors\"\n",
        "\n",
        "elif model_name == \"SukiyakiMix-v1.0\":\n",
        "  url1  = \"https://huggingface.co/Vsukiyaki/SukiyakiMix-v1.0/resolve/main/SukiyakiMix-v1.0-fp16.safetensors\" \n",
        "  path1 = \"SukiyakiMix-v1.0-fp16.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v1.1\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v1/7th_anime_v1.1.safetensors\"\n",
        "  path1  = \"7th_anime_v1.1.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v2_A\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_A.safetensors\"\n",
        "  path1  = \"7th_anime_v2_A.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v2_B\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_B.safetensors\"\n",
        "  path1  = \"7th_anime_v2_B.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v2_C\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_C.safetensors\"\n",
        "  path1  = \"7th_anime_v2_C.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v2_G\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_G.safetensors\"\n",
        "  path1  = \"7th_anime_v2_G.safetensors\"\n",
        "  \n",
        "elif model_name == \"7th_anime_v3_A\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_A.safetensors\" \n",
        "  path1 = \"7th_anime_v3_A.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v3_B\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_B.safetensors\"\n",
        "  path1  = \"7th_anime_v3_B.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_v3_C\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_C.safetensors\"\n",
        "  path1  = \"7th_anime_v3_C.safetensors\"\n",
        "\n",
        "elif model_name == \"7th_anime_3.1_B\":\n",
        "  url1  = \"https://huggingface.co/syaimu/7th_test/resolve/main/7th_anime_3.1_B.ckpt\" \n",
        "  path1 = \"7th_anime_3.1_B.ckpt\"\n",
        "\n",
        "elif model_name == \"YuzuGinger_v4\":\n",
        "  url1  = \"https://huggingface.co/thiros/YuzuLemonTea/resolve/main/YuzuGinger_v4-pruned.safetensors\" \n",
        "  path1 = \"YuzuGinger_v4-pruned.safetensors\"\n",
        "\n",
        "elif model_name == \"YuzuLemonMilk_v1.5\":\n",
        "  url1  = \"https://huggingface.co/thiros/YuzuLemonTea/resolve/main/YuzuLemonMilk_v1.5-pruned.safetensors\" \n",
        "  path1 = \"YuzuLemonMilk_v1.5-pruned.safetensors\"\n",
        "\n",
        "elif model_name == \"Defmix-v1.1\":\n",
        "  url1  = \"https://huggingface.co/Defpoint/Defmix-v1.0/resolve/main/Defmix-v1.1.safetensors\" \n",
        "  path1 = \"Defmix-v1.1.safetensors\"\n",
        "\n",
        "elif model_name == \"RDtMix\":\n",
        "  url1  = \"https://huggingface.co/Hemlok/DateMix/resolve/main/RDtMix-fp16.safetensors\" \n",
        "  path1 = \"Defmix-fictio.safetensors\"\n",
        "\n",
        "elif model_name == \"meadmix\":\n",
        "  url1  = \"https://huggingface.co/sazanka-imoto/MeadMix/resolve/main/meadmix.safetensors\" \n",
        "  path1 = \"meadmix.safetensors\"\n",
        "\n",
        "elif model_name == \"Nabylon-v1.0\":\n",
        "  url1  = \"https://huggingface.co/NegiInNattoMaki/Nabylon-v1.0/resolve/main/Nabylon-v1.0-fp16.safetensors\" \n",
        "  path1 = \"Nabylon-v1.0-fp16.safetensors\"\n",
        "\n",
        "elif model_name == \"LonganMix\":\n",
        "  url1  = \"https://huggingface.co/Hemlok/LonganMix/resolve/main/LonganMix-fp16.ckpt\" \n",
        "  path1 = \"LonganMix-fp16.ckpt\" \n",
        "\n",
        "elif model_name == \"Corneo's 7th Heaven Mix\":\n",
        "  url1  = \"https://civitai.com/api/download/models/5338\" \n",
        "  path1 = \"Corneos_7th_Heaven_Mix.safetensors\"\n",
        "\n",
        "elif model_name == \"Corneo's Shinra26 Mix\":\n",
        "  url1  = \"https://civitai.com/api/download/models/6918\" \n",
        "  path1 = \"Corneos_Shinra26_Mix.safetensors\"\n",
        "\n",
        "elif model_name == \"atwmix_test2\":\n",
        "  url1  = \"https://huggingface.co/atsuwo/atwmix_test2/resolve/main/atwmix_test2.ckpt\" \n",
        "  path1 = \"atwmix_test2.ckpt\"\n",
        "\n",
        "elif model_name == \"Graham_Special\":\n",
        "  url1  = \"https://huggingface.co/Graham-san/Graham_Special/resolve/main/Graham%20Special%20v1.ckpt\" \n",
        "  path1 = \"Graham_Special.ckpt\"\n",
        "\n",
        "elif model_name == \"BalorV2featAbyssOrange2\":\n",
        "  url1  = \"https://huggingface.co/ploughB660/Balor-V2/resolve/main/BalorV2featAbyssOrange2.safetensors\" \n",
        "  path1 = \"BalorV2featAbyssOrange2.safetensors\"\n",
        "\n",
        "elif model_name == \"Balor-V2.5featAOM3A3\":\n",
        "  url1  = \"https://huggingface.co/ploughB660/Balor-V3/resolve/main/Balor-V2.5featAOM3A3.safetensors\"\n",
        "  path1 = \"Balor-V2.5featAOM3A3.safetensors\"\n",
        "\n",
        "\n",
        "elif model_name == \"ChilloutMix\":\n",
        "  url1  = \"https://civitai.com/api/download/models/11745\"\n",
        "  path1 = \"ChilloutMix.safetensors\"\n",
        "\n",
        "elif model_name == \"wd-1-5-beta2\":\n",
        "  url1  = \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp16.safetensors\"\n",
        "  path1 = \"wd-1-5-beta2-aesthetic-fp16.safetensors\"\n",
        "\n",
        "else:\n",
        "  print(\"Invalid model name.\")\n",
        "\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M $url1 -d /content/stable-diffusion-webui/models/Stable-diffusion -o $path1\n",
        " "
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@title 3.VAE\n",
        "\n",
        "vae_name =\"Anything-V3.0.vae.pt\" #@param [\"vae-ft-mse-840000-ema-pruned\",\"vae-ft-mse-560000-ema-pruned\",\"kl-f8-anime\",\"kl-f8-anime2\",\"Anything-V3.0.vae.pt\",\"autoencoder_fix_kl-f8-trinart_characters\",\"orangemix.vae.pt\"] {allow-input: false}\n",
        "\n",
        "if vae_name == \"vae-ft-mse-840000-ema-pruned\":\n",
        "  url2  = \"https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors\" \n",
        "  path2 = \"vae-ft-mse-840000-ema-pruned.vae.pt\"\n",
        "\n",
        "elif vae_name == \"vae-ft-mse-560000-ema-pruned\":\n",
        "  url2  = \"https://huggingface.co/stabilityai/sd-vae-ft-ema-original/resolve/main/vae-ft-ema-560000-ema-pruned.ckpt\" \n",
        "  path2 = \"vae-ft-ema-560000-ema-pruned.vae.pt\"\n",
        "\n",
        "elif vae_name == \"kl-f8-anime\":\n",
        "  url2  = \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime.ckpt\" \n",
        "  path2 = \"kl-f8-anime.vae.pt\"\n",
        "\n",
        "elif vae_name == \"kl-f8-anime2\":\n",
        "  url2  = \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt\" \n",
        "  path2 = \"kl-f8-anime2.vae.pt\"\n",
        "\n",
        "elif vae_name == \"Anything-V3.0.vae.pt\":\n",
        "  url2  = \"https://huggingface.co/ckpt/anything-v3.0/resolve/main/Anything-V3.0.vae.pt\" \n",
        "  path2 = \"Anything-V3.0.vae.pt\"\n",
        "\n",
        "elif vae_name == \"autoencoder_fix_kl-f8-trinart_characters\":\n",
        "  url2  = \"https://huggingface.co/naclbit/trinart_characters_19.2m_stable_diffusion_v1/resolve/main/autoencoder_fix_kl-f8-trinart_characters.ckpt\" \n",
        "  path2 = \"autoencoder_fix_kl-f8-trinart_characters.vae.pt\"\n",
        "\n",
        "elif vae_name == \"orangemix.vae.pt\":\n",
        "  url2  = \"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt\" \n",
        "  path2 = \"orangemix.vae.pt\"\n",
        "\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M $url2 -d /content/stable-diffusion-webui/models/Stable-diffusion -o $path2"
      ],
      "metadata": {
        "id": "hGBWj7GV3xnx",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title 追加学習モデル LoRA・Textual Inversion･Lycoris（任意）\n",
        "\n",
        "Lora_name =\"None\" #@param [\"None\",\"Anime_Screencap_Style_LoRA\",\"One_Piece_Wano_Style_LoRA\",\"Eula_LoRA\",\"Yelan_LoRA\",\"Barbara_LoRA\",\"Diluc_LoRA\",\"Mona_LoRA\",\"Ganyu_LoRA\",\"Jean_LoRA\",\"Kujou_Sara_LoRA\",\"Lisa_LoRA\",\"Raiden_Shogun_LoRA\",\"Shenhe_LoRA\",\"Yoimiya_LoRA\",\"Zhongli_LoRA\",\"Yae_Miko_LoRA\"] {allow-input:false}\n",
        "\n",
        "if Lora_name == \"None\":\n",
        "  url3  = None \n",
        "  path3 = None\n",
        "\n",
        "elif Lora_name == \"Anime_Screencap_Style_LoRA\":\n",
        "  url3  = \"https://civitai.com/api/download/models/5742\" \n",
        "  path3 = \"Anime_Screencap_Style_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"One_Piece_Wano_Style_LoRA\":\n",
        "  url3  = \"https://civitai.com/api/download/models/6331\" \n",
        "  path3 = \"One_Piece_Wano_Style_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Eula_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Eula/eulaMedium.safetensors\" \n",
        "  path3 = \"Eula_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Yelan_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Yelan/Yelan_Hard.safetensors\" \n",
        "  path3 = \"Yelan_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Barbara_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Barbara/Barbara_mediumpruned.safetensors\" \n",
        "  path3 = \"Barbara_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Diluc_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Diluc/Diluc_Mediumpruned.safetensors\" \n",
        "  path3 = \"Diluc_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Mona_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Mona/Mona_mediumpruned.safetensors\" \n",
        "  path3 = \"Mona_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Ganyu_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Ganyu/Ganyu_Medium.safetensors\" \n",
        "  path3 = \"Ganyu_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Jean_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Jean/Jean_Medium.safetensors\" \n",
        "  path3 = \"Jean_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Kujou_Sara_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Kujou%20Sara/kujou%20sara_Heavy.safetensors\" \n",
        "  path3 = \"kujou_sara_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Lisa_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Lisa/Lisa_Medium.safetensors\" \n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Lisa_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Raiden_Shogun_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Raiden%20Shogun/raiden%20shogun_LoRA.safetensors\"\n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Raiden_Shogun_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Shenhe_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Shenhe/Shenhe_Medium.safetensors\"\n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Shenhe_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Yoimiya_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Yoimiya/Yoimiya_mediumpruned.safetensors\"\n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Yoimiya_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Zhongli_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Zhongli/Zhongli_Medium.safetensors\"\n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Zhongli_LoRA.safetensors\"\n",
        "\n",
        "elif Lora_name == \"Yae_Miko_LoRA\":\n",
        "  url3  = \"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/yae%20miko/yae%20miko_Mediumpruned.safetensors\"\n",
        "  path3 = \"/content/stable-diffusion-webui/models/Lora/Yae_Miko_LoRA.safetensors\"\n",
        "\n",
        "else:\n",
        "  print(\"Invalid Lora name.\")\n",
        "\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M $url3 -d /content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora -o $path3\n",
        "\n",
        "Textual_name =\"None\"#@param [\"None\",\"Corneo's Makima\",\"Corneo_Power\",\"Yor Forger\",\"Corneo's POV Bound Wrists Missionary\",\"Corneo's Spitroast Threesome\",\"Corneo's Anal sex\",\"Corneo's Side-view Deepthroat\",\"Corneo's Covering Breasts -Two Hands\",\"Corneo's X-shaped Nipple Pasties\",\"Corneo's POV Oral\",\"Corneo's POV Paizuri\",\"Corneo's Covering Breasts -One Arm\",\"Corneo's Arm Grab Doggystyle\",\"Corneo's Covering Breasts -Arms Crossed\",\"Licking Oral\"] {allow-input:false}\n",
        "\n",
        "if Textual_name == \"None\":\n",
        "  url4  = None \n",
        "  path4 = None\n",
        "\n",
        "if Textual_name == \"Corneo's Makima\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6197\"\n",
        "  path4 = \"Corneo_Makima.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo_Power\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6027\" \n",
        "  path4 = \"Corneo_Power.pt\"\n",
        "\n",
        "elif Textual_name == \"Yor Forger\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6300\" \n",
        "  path4 = \"Corneo_POV.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's POV Bound Wrists Missionary\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5405\" \n",
        "  path4 = \"Corneo_POV.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Spitroast Threesome\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6770\" \n",
        "  path4 = \"Corneo_Spitroast_Threesome.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Anal sex\":\n",
        "  url4  = \"https://civitai.com/api/download/models/4597\" \n",
        "  path4 = \"Corneo_Anal.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Side-view Deepthroat\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6242\" \n",
        "  path4 = \"Corneo_Side-view_Deepthroat.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Covering Breasts -Two Hands\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6033\" \n",
        "  path4 = \"Corneo_Covering_Breasts.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's X-shaped Nipple Pasties\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6318\" \n",
        "  path4 = \"Corneo_Nipple_Pasties.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's POV Oral\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5051\" \n",
        "  path4 = \"Corneo_POV_Oral.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's POV Paizuri\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5066\" \n",
        "  path4 = \"Corneo_POV_Paizuri.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Covering Breasts -One Arm\":\n",
        "  url4  = \"https://civitai.com/api/download/models/6034\" \n",
        "  path4 = \"Corneo_Covering_Breasts-One.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Arm Grab Doggystyle\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5160\" \n",
        "  path4 = \"Corneo_Arm_Grab_Doggystyle.pt\"\n",
        "\n",
        "elif Textual_name == \"Corneo's Covering Breasts -Arms Crossed\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5746\" \n",
        "  path4 = \"Corneo_Covering_Breasts-Arms_Crossed.pt\"\n",
        "\n",
        "elif Textual_name == \"Licking Oral\":\n",
        "  url4  = \"https://civitai.com/api/download/models/5746\" \n",
        "  path4 = \"Licking_Oral.pt\"\n",
        "\n",
        "else:\n",
        "  print(\"Invalid model name.\")\n",
        "\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M $url4 -d /content/stable-diffusion-webui/embeddings -o $path4\n",
        "\n",
        "#Lycoris-LoRA\n",
        "Lycoris_name =\"None\"#@param [\"None\"] {allow-input:false}\n",
        "#準備中"
      ],
      "metadata": {
        "id": "QDG0oyNU6R1L",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title  4.WebUIの起動\n",
        "!sed -i -e '''/    prepare_environment()/a\\    os.system\\(f\\\"\"\"sed -i -e ''\\\"s/dict()))/dict())).cuda()/g\\\"'' /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py\"\"\")''' /content/stable-diffusion-webui/launch.py\n",
        "!sed -i -e 's/fastapi==0.94.0/fastapi==0.89.1/g' /content/stable-diffusion-webui/requirements_versions.txt\n",
        "\n",
        "!mkdir /content/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui/models\n",
        "\n",
        "!python launch.py --listen --xformers --enable-insecure-extension-access --theme dark --gradio-queue --multiple"
      ],
      "metadata": {
        "id": "Pet8RD2Q6-4L",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "▼学習済みモデルの検索<br>\n",
        "[HuggingFace](https://huggingface.co/ckpt)<br>\n",
        "[Civitai](https://civitai.com/)<br><br>\n",
        "▼Colabページの説明<br>\n",
        "[stable-diffusion-webui-colab-japanese](https://github.com/AI-Nios/stable-diffusion-webui-colab-japanese)<br>\n",
        "<br>\n",
        "▼以前のバージョン<br>\n",
        "[Nios_v2.ipynb](https://colab.research.google.com/github/AI-Nios/stable-diffusion-webui-colab-japanese/blob/main/Nios_v2_update.ipynb)<br><br> \n",
        "▼記事<br>\n",
        "[Google Colabで画像生成AIの色々なモデルを簡単に使おう](https://note.com/ai_nios/n/n0826a33edc8a)<br>\n",
        "<br>\n",
        "▼ColabでLoRA学習<br>\n",
        "[Linaqruf/kohya-trainer](https://github.com/Linaqruf/kohya-trainer)\n",
        "<br>\n",
        "[あかちゃんLoRAノートブック](https://seesaawiki.jp/nai_ch/d/kohya_train_network_simple)<br>\n",
        "もしくはこちら<br>\n",
        "[【最新バージョン】Loraモデル学習をGoogle Colabで作る方法解説。Kohya LoRA Dreambooth v14.1使用。【Stable Diffusion】](https://www.youtube.com/watch?v=ddpaECgRCHg&ab_channel=ShinanoMatsumoto%E3%83%BB%E6%99%B4%E3%82%8C%E6%99%82%E3%80%85%E3%82%AC%E3%82%B8%E3%82%A7%E3%83%83%E3%83%88)<br>\n",
        "もしくはこちら<br>\n",
        "[【StableDiffusion】初めてのAIお絵かき講座 その4 キャラや絵柄を追加学習！LoRA編【Anything V4.5】](https://www.nicovideo.jp/watch/sm41895651)\n"
      ],
      "metadata": {
        "id": "4iNjTp_bGOpO"
      }
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "provenance": [],
      "include_colab_link": true
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}