Nios_v3.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_v3.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "2023/4/21 <br>\n",
        "有料プランの方のみご利用いただけます。<br>\n",
        "[詳細](https://research.google.com/colaboratory/faq.html#limitations-and-restrictions)"
      ],
      "metadata": {
        "id": "HT3GILQvO6QR"
      }
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "3rvLOgqyDUdD",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title 1.WebUIのインストール\n",
        "import os\n",
        "from google.colab import drive\n",
        "\n",
        "#ドライブをマウント\n",
        "Mount_Drive = False # @param {type: \"boolean\"}\n",
        "\n",
        "if Mount_Drive:\n",
        "    if not os.path.exists(\"/content/drive\"):\n",
        "        drive.mount(\"/content/drive\")\n",
        "\n",
        "%cd /content\n",
        "\n",
        "!apt -y update -qq\n",
        "!apt -y install -qq aria2\n",
        "\n",
        "!pip install -q torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 torchtext==0.15.1 torchdata==0.6.0 --extra-index-url https://download.pytorch.org/whl/cu118 -U\n",
        "!pip install -q xformers==0.0.19 triton==2.0.0 -U\n",
        "\n",
        "#WebUIのインストール\n",
        "#!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui\n",
        "#!git clone -b v2.1 https://github.com/camenduru/stable-diffusion-webui\n",
        "!git clone -b v2.2 https://github.com/camenduru/stable-diffusion-webui\n",
        "\n",
        "#Extensions1 絶対入れておくべき\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/kohya-ss/sd-webui-additional-networks /content/stable-diffusion-webui/extensions/sd-webui-additional-networks\n",
        "!git clone https://github.com/yankooliveira/sd-webui-photopea-embed.git /content/stable-diffusion-webui/extensions/sd-webui-photopea-embed\n",
        "\n",
        "#Extensions1.5\n",
        "!git clone https://huggingface.co/embed/negative /content/stable-diffusion-webui/embeddings/negative\n",
        "!git clone https://huggingface.co/embed/lora /content/stable-diffusion-webui/models/Lora/positive\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/embed/upscale/resolve/main/4x-UltraSharp.pth -d /content/stable-diffusion-webui/models/ESRGAN -o 4x-UltraSharp.pth\n",
        "!git clone https://github.com/camenduru/sd-webui-tunnels /content/stable-diffusion-webui/extensions/sd-webui-tunnels\n",
        "!wget https://raw.githubusercontent.com/camenduru/stable-diffusion-webui-scripts/main/run_n_times.py -O /content/stable-diffusion-webui/scripts/run_n_times.py\n",
        "\n",
        "#Extensions2　任意\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/opparco/stable-diffusion-webui-two-shot.git /content/stable-diffusion-webui/extensions/stable-diffusion-webui-two-shot\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",
        "    !git clone https://github.com/hnmr293/sd-webui-llul /content/stable-diffusion-webui/extensions/sd-webui-llul\n",
        "    !git clone https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git /content/stable-diffusion-webui/extensions/Stable-Diffusion-Webui-Civitai-Helper\n",
        "    !git clone https://github.com/hnmr293/sd-webui-cutoff /content/stable-diffusion-webui/extensions/sd-webui-cutoff\n",
        "    !git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui-wildcards.git /content/stable-diffusion-webui/extensions/stable-diffusion-webui-wildcards\n",
        "    !git clone https://github.com/etherealxx/batchlinks-webui /content/stable-diffusion-webui/extensions/batchlinks-webui\n",
        "    !git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg /content/stable-diffusion-webui/extensions/stable-diffusion-webui-rembg\n",
        "    !git clone https://github.com/blue-pen5805/sdweb-easy-prompt-selector.git /content/stable-diffusion-webui/extensions/sdweb-easy-prompt-selector\n",
        "%cd /content/stable-diffusion-webui\n",
        "!git reset --hard\n",
        "!git -C /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai reset --hard\n",
        "\n",
        "#Extensions3(ControlNet)　任意\n",
        "ControlNet = False #@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",
        "    !git clone https://github.com/hnmr293/posex /content/stable-diffusion-webui/extensions/posex\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_ip2p_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_shuffle_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_canny_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_depth_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_depth_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_inpaint_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_lineart_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_mlsd_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_normalbae_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_openpose_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_scribble_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_seg_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_softedge_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15s2_lineart_anime_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11u_sd15_tile_fp16.safetensors -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11u_sd15_tile_fp16.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_ip2p_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_shuffle_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_canny_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_depth_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_depth_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_inpaint_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_lineart_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_mlsd_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_normalbae_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_openpose_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_scribble_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_seg_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_softedge_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15s2_lineart_anime_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11u_sd15_tile_fp16.yaml -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11u_sd15_tile_fp16.yaml\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/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-v1-1/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-v1-1/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-v1-1/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-v1-1/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-v1-1/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-v1-1/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-v1-1/resolve/main/t2iadapter_canny_sd14v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_canny_sd14v1.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/t2iadapter_canny_sd15v2.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_canny_sd15v2.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/t2iadapter_depth_sd15v2.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_depth_sd15v2.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/t2iadapter_sketch_sd15v2.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_sketch_sd15v2.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/t2iadapter_zoedepth_sd15v1.pth -d /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_zoedepth_sd15v1.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",
        "import os\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\",\"AniReality-Mix\"] {allow-input: false}\n",
        "\n",
        "model_dict={\n",
        "    \"stable_diffusion_v1.5\":\"https://huggingface.co/camenduru/sd15/resolve/main/v1-5-pruned-emaonly.ckpt\",\n",
        "    \"stable_diffusion_v2.1\":\"https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.ckpt\",\n",
        "    \"waifu_diffusion_v1.3\":\"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt\",\n",
        "    \"Openjourney\":\"https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.safetensors\",\n",
        "    \"anything-v3.0\":\"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/anything-v3-fp32-pruned.safetensors\",\n",
        "    \"anything-v4.0\":\"https://huggingface.co/ckpt/anything-v4.0/resolve/main/anything-v4.0-pruned.ckpt\",\n",
        "    \"anything-v4.5\":\"https://huggingface.co/ckpt/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors\",\n",
        "    \"Counterfeit-V2.0\":\"https://huggingface.co/gsdf/Counterfeit-V2.0/resolve/main/Counterfeit-V2.0fp16.safetensors\",\n",
        "    \"Counterfeit-V2.5\":\"https://huggingface.co/gsdf/Counterfeit-V2.5/resolve/main/Counterfeit-V2.5_pruned.safetensors\",\n",
        "    \"8528-diffusion\":\"https://huggingface.co/852wa/8528-diffusion/resolve/main/8528d-final.ckpt\",\n",
        "    \"AbyssOrangeMix2\":\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_nsfw.safetensors\",\n",
        "    \"EerieOrangeMix2\":\"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/EerieOrangeMix/EerieOrangeMix2.safetensors\",\n",
        "    \"trinart2_step115000\":\"https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step115000.ckpt\",\n",
        "    \"EimisAnimeDiffusion_1.0v\":\"https://huggingface.co/eimiss/EimisAnimeDiffusion_1.0v/resolve/main/EimisAnimeDiffusion_1-0v.ckpt\",\n",
        "    \"dreamlike-photoreal-2.0\":\"https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0/resolve/main/dreamlike-photoreal-2.0.safetensors\",\n",
        "    \"Cyberpunk-Anime-Diffusion\":\"https://huggingface.co/DGSpitzer/Cyberpunk-Anime-Diffusion/resolve/main/Cyberpunk-Anime-Diffusion.ckpt\",\n",
        "    \"Plat_Diffusion_v1.3.1\":\"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\",\n",
        "    \"ACertainThing\":\"https://huggingface.co/JosephusCheung/ACertainThing/resolve/main/ACertainThing.ckpt\",\n",
        "    \"pastel_mix\":\"https://huggingface.co/andite/pastel-mix/resolve/main/pastelmix-better-vae-fp16.safetensors\",\n",
        "    \"Basil_mix_fixed\":\"https://huggingface.co/nuigurumi/basil_mix/resolve/main/Basil_mix_fixed.safetensors\",\n",
        "    \"Realistic_Vision_V1.3\":\"https://civitai.com/api/download/models/6987\",\n",
        "    \"SukiyakiMix-v1.0\":\"https://huggingface.co/Vsukiyaki/SukiyakiMix-v1.0/resolve/main/SukiyakiMix-v1.0-fp16.safetensors\",\n",
        "    \"7th_anime_v1.1\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v1/7th_anime_v1.1.safetensors\",\n",
        "    \"7th_anime_v2_A\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_A.safetensors\",\n",
        "    \"7th_anime_v2_B\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_B.safetensors\",\n",
        "    \"7th_anime_v2_C\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_C.safetensors\",\n",
        "    \"7th_anime_v2_G\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v2/7th_anime_v2_G.safetensors\",\n",
        "    \"7th_anime_v3_A\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_A.safetensors\",\n",
        "    \"7th_anime_v3_B\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_B.safetensors\",\n",
        "    \"7th_anime_v3_C\":\"https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_C.safetensors\",\n",
        "    \"7th_anime_3.1_B\":\"https://huggingface.co/syaimu/7th_test/resolve/main/7th_anime_3.1_B.ckpt\",\n",
        "    \"YuzuGinger_v4\":\"https://huggingface.co/thiros/YuzuLemonTea/resolve/main/YuzuGinger_v4-pruned.safetensors\",\n",
        "    \"YuzuLemonMilk_v1.5\":\"https://huggingface.co/thiros/YuzuLemonTea/resolve/main/YuzuLemonMilk_v1.5-pruned.safetensors\",\n",
        "    \"Defmix-v1.1\":\"https://huggingface.co/Defpoint/Defmix-v1.0/resolve/main/Defmix-v1.1.safetensors\",\n",
        "    \"RDtMix\":\"https://huggingface.co/Hemlok/DateMix/resolve/main/RDtMix-fp16.safetensors\",\n",
        "    \"meadmix\":\"https://huggingface.co/sazanka-imoto/MeadMix/resolve/main/meadmix.safetensors\",\n",
        "    \"Nabylon-v1.0\":\"https://huggingface.co/NegiInNattoMaki/Nabylon-v1.0/resolve/main/Nabylon-v1.0-fp16.safetensors\",\n",
        "    \"LonganMix\":\"https://huggingface.co/Hemlok/LonganMix/resolve/main/LonganMix-fp16.ckpt\",\n",
        "    \"Corneo's_7th_Heaven_Mix\":\"https://civitai.com/api/download/models/5338\",\n",
        "    \"Corneo's_Shinra26_Mix\":\"https://civitai.com/api/download/models/6918\",\n",
        "    \"atwmix_test2\":\"https://huggingface.co/atsuwo/atwmix_test2/resolve/main/atwmix_test2.ckpt\",\n",
        "    \"Graham_Special\":\"https://huggingface.co/Graham-san/Graham_Special/resolve/main/Graham%20Special%20v1.ckpt\",\n",
        "    \"BalorV2featAbyssOrange2\":\"https://huggingface.co/ploughB660/Balor-V2/resolve/main/BalorV2featAbyssOrange2.safetensors\",\n",
        "    \"Balor-V2.5featAOM3A3\":\"https://huggingface.co/ploughB660/Balor-V3/resolve/main/Balor-V2.5featAOM3A3.safetensors\",\n",
        "    \"ChilloutMix\":\"https://civitai.com/api/download/models/11745\",\n",
        "    \"wd-1-5-beta2\":\"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp16.safetensors\",\n",
        "    \"AniReality-Mix\":\"https://civitai.com/api/download/models/33305\"\n",
        "}\n",
        "\n",
        "model_url = model_dict.get(model_name)\n",
        "\n",
        "def install(model_url):\n",
        "    base_name = os.path.basename(model_url)\n",
        "\n",
        "    if model_url.startswith(\"https://drive.google.com\"):\n",
        "        os.chdir(\"/content/stable-diffusion-webui/models/Stable-diffusion\")\n",
        "        !gdown --fuzzy {model_url}\n",
        "    elif model_url.startswith(\"https://huggingface.co/\"):\n",
        "        if \"/blob/\" in model_url:\n",
        "            model_url = model_url.replace(\"/blob/\", \"/resolve/\")\n",
        "        huggingface_token =\"YOUR_HUGGINGFACE_TOKEN\" # 自分のトークンに変更\n",
        "        user_header = f'\"Authorization: Bearer {huggingface_token}\"'\n",
        "        !aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -s 16 -k 1M {model_url} -d /content/stable-diffusion-webui/models/Stable-diffusion -o {base_name}\n",
        "    else:\n",
        "        !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d /content/stable-diffusion-webui/models/Stable-diffusion {model_url}\n",
        "\n",
        "install(model_url)"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@title 3.VAE\n",
        "import os\n",
        "\n",
        "vae_name = \"orangemix.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\",\"anything-v4.0.vae.pt\",\"autoencoder_fix_kl-f8-trinart_characters\",\"orangemix.vae.pt\"] {allow-input: false}\n",
        "\n",
        "url_dict = {\n",
        "    \"vae-ft-mse-840000-ema-pruned\": \"https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors\",\n",
        "    \"vae-ft-mse-560000-ema-pruned\": \"https://huggingface.co/stabilityai/sd-vae-ft-ema-original/resolve/main/vae-ft-ema-560000-ema-pruned.ckpt\",\n",
        "    \"kl-f8-anime\": \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime.ckpt\",\n",
        "    \"kl-f8-anime2\": \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt\",\n",
        "    \"Anything-V3.0.vae.pt\": \"https://huggingface.co/ckpt/anything-v3.0/resolve/main/Anything-V3.0.vae.pt\",\n",
        "    \"anything-v4.0.vae.pt\":\"https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt\",\n",
        "    \"autoencoder_fix_kl-f8-trinart_characters\": \"https://huggingface.co/naclbit/trinart_characters_19.2m_stable_diffusion_v1/resolve/main/autoencoder_fix_kl-f8-trinart_characters.ckpt\",\n",
        "    \"orangemix.vae.pt\": \"https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt\",\n",
        "}\n",
        "\n",
        "vae_url = url_dict.get(vae_name)\n",
        "\n",
        "def install(vae_url):\n",
        "    base_name = os.path.basename(vae_url)\n",
        "    !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 {vae_url} -d /content/stable-diffusion-webui/models/VAE -o {base_name}\n",
        "    print(base_name)\n",
        "install(vae_url)\n"
      ],
      "metadata": {
        "id": "hGBWj7GV3xnx",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title カスタムモデルまたはExtensionsの追加(任意)\n",
        "#@markdown [実行方法](https://twitter.com/ai_nios/status/1639659834240008192?s=20)\n",
        "import os\n",
        "\n",
        "model_type =\"LoRA\" #@param [\"Checkpoint\",\"Hypernetwork\",\"Texual_Inversion\",\"Aesthetic_Gradients\",\"LoRA\",\"LoRA(additional-networks)\",\"LyCORIS\",\"Wildcards\", \"VAE\",\"Webui_Extension\",\"Easy_Prompt_Selector\"] {allow-input: false}\n",
        "\n",
        "if model_type == \"Checkpoint\":\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/models/Stable-diffusion\"\n",
        "elif model_type == \"Hypernetwork\":\n",
        "    !git https://github.com/antis0007/sd-webui-multiple-hypernetworks.git /content/stable-diffusion-webui/extensions/sd-webui-multiple-hypernetworks\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/extensions/sd-webui-multiple-hypernetworks\"\n",
        "elif model_type ==\"Aesthetic_Gradients\":\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/embeddings\"\n",
        "elif model_type == \"LoRA\":\n",
        "    !mkdir /content/stable-diffusion-webui/models/Lora\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/models/Lora\"\n",
        "elif model_type == \"LoRA(additional-networks)\":\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora\"\n",
        "elif model_type == \"LyCORIS\":\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora\"\n",
        "\n",
        "modelUrl = \"\"  # @param {'type': 'string'}\n",
        "\n",
        "if model_type == \"Wildcards\":\n",
        "    !git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui-wildcards.git /content/stable-diffusion-webui/extensions/stable-diffusion-webui-wildcards\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/extensions/stable-diffusion-webui-wildcards/wildcards\"\n",
        "    !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {pretrained_model} {url}\n",
        "\n",
        "elif model_type == \"Webui_Extension\":\n",
        "    def install(url):\n",
        "        import os\n",
        "        base_name = os.path.basename(url)\n",
        "        print(\"base_name\")\n",
        "        pretrained_model = \"/content/stable-diffusion-webui/extensions\"\n",
        "        !git clone {url} {pretrained_model}/{base_name}\n",
        "    install(modelUrl)\n",
        "\n",
        "elif model_type == \"Easy_Prompt_Selector\":\n",
        "    pretrained_model = \"/content/stable-diffusion-webui/extensions/sdweb-easy-prompt-selector/tags\"\n",
        "    if modelUrl == \"\":\n",
        "        from google.colab import files\n",
        "        import os\n",
        "\n",
        "        def upload_files():\n",
        "          uploaded = files.upload()\n",
        "          for filename in uploaded.keys():\n",
        "            if filename.endswith(\".yml\") or filename.endswith(\".yaml\"):\n",
        "              os.rename(filename, \"/content/stable-diffusion-webui/extensions/sdweb-easy-prompt-selector/tags/\" + filename)\n",
        "              print(\"Uploaded file:\", filename)\n",
        "\n",
        "        upload_files()\n",
        "\n",
        "    else:\n",
        "        def install(url):\n",
        "            base_name = os.path.basename(url)\n",
        "\n",
        "            if url.startswith(\"https://drive.google.com\"):\n",
        "                os.chdir(\"/content/stable-diffusion-webui/extensions/sdweb-easy-prompt-selector/tags\")\n",
        "                !gdown --fuzzy {url}\n",
        "            elif url.startswith(\"https://huggingface.co/\"):\n",
        "                if \"/blob/\" in url:\n",
        "                    url = url.replace(\"/blob/\", \"/resolve/\")\n",
        "                huggingface_token =\"YOUR_HUGGINGFACE_TOKEN\" # 自分のトークンに変更\n",
        "                user_header = f'\"Authorization: Bearer {huggingface_token}\"'\n",
        "                !aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -s 16 -k 1M {url} -d {pretrained_model} -o {base_name}\n",
        "            else:\n",
        "                !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {pretrained_model} {url}\n",
        "\n",
        "        install(modelUrl)\n",
        "\n",
        "elif modelUrl == \"\":\n",
        "  print(\"modelUrlに値が指定されていないため、スキップされます。\")\n",
        "\n",
        "else:\n",
        "    def install(url):\n",
        "        base_name = os.path.basename(url)\n",
        "        print(base_name)\n",
        "        if url.startswith(\"https://drive.google.com\"):\n",
        "            os.chdir(pretrained_model)\n",
        "            !gdown --fuzzy {url}\n",
        "        elif url.startswith(\"https://huggingface.co/\"):\n",
        "                    #@markdown ▼Hugging Faceのプライベートモデルを使用する場合、[アクセストークン](https://huggingface.co/settings/tokens)を入力してください。\n",
        "            huggingface_token = \"\"  # @param {type:\"string\"}\n",
        "            user_header = f'\"Authorization: Bearer {huggingface_token}\"'\n",
        "            if \"/blob/\" in url:\n",
        "                url = url.replace(\"/blob/\", \"/resolve/\")\n",
        "            if model_type == \"VAE\":\n",
        "                pretrained_model = \"/content/stable-diffusion-webui/models/Stable-diffusion\"\n",
        "                if \".ckpt\" in base_name or \".safetensors\" in base_name:\n",
        "                    base_name = base_name.replace(\".ckpt\", \".vae.pt\").replace(\".safetensors\", \".vae.pt\")\n",
        "                    print(base_name)\n",
        "            elif model_type ==\"Texual_Inversion\":\n",
        "                if \".ckpt\" in base_name or \".safetensors\" in base_name:\n",
        "                    base_name = base_name.replace(\".ckpt\", \".pt\").replace(\".safetensors\", \".pt\")\n",
        "                    print(base_name)\n",
        "                pretrained_model = \"/content/stable-diffusion-webui/embeddings\"\n",
        "\n",
        "            !aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 -d {pretrained_model} -o {base_name} {url}\n",
        "\n",
        "        else:\n",
        "            if model_type == \"VAE\":\n",
        "                pretrained_model = \"/content/stable-diffusion-webui/models/Stable-diffusion\"\n",
        "                if \".ckpt\" in base_name or \".safetensors\" in base_name:\n",
        "                    base_name = base_name.replace(\".ckpt\", \".vae.pt\").replace(\".safetensors\", \".vae.pt\")\n",
        "                    print(base_name)\n",
        "            elif model_type ==\"Texual_Inversion\":\n",
        "                if \".ckpt\" in base_name or \".safetensors\" in base_name:\n",
        "                    base_name = base_name.replace(\".ckpt\", \".pt\").replace(\".safetensors\", \".pt\")\n",
        "                    print(base_name)\n",
        "                pretrained_model = \"/content/stable-diffusion-webui/embeddings\"\n",
        "            !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {pretrained_model} {url}\n",
        "\n",
        "    install(modelUrl)\n",
        "\n",
        "#@markdown ▼driveからファイルをコピーする場合、[ファイルパス](https://camo.qiitausercontent.com/76671e92e7da4f86f522fcb372fb49db27fa805b/68747470733a2f2f71696974612d696d6167652d73746f72652e73332e61702d6e6f727468656173742d312e616d617a6f6e6177732e636f6d2f302f3533333834342f66623462326238382d383938302d373932362d353063632d3063613830386137343334642e706e67)を入力してください。ファイルが見つからない場合はスキップされます。\n",
        "drivePath = \"\\u4F8B:/content/drive/MyDrive/LoRA/iroha.safetensors\" #@param {type: \"string\"}\n",
        "\n",
        "if drivePath:\n",
        "  if \"例:\" in drivePath:\n",
        "    print(\"\")\n",
        "  else:\n",
        "    import os\n",
        "    from google.colab import drive\n",
        "    drive.mount(\"/content/drive\")\n",
        "    # check if file exists in drivePath\n",
        "    if not os.path.exists(drivePath):\n",
        "      # check if file exists with \"/\" prefix\n",
        "      if not os.path.exists(\"/\" + drivePath):\n",
        "        # check if file exists with \"/content/drive/MyDrive/\" prefix\n",
        "        if not os.path.exists(\"/content/drive/MyDrive/\" + drivePath):\n",
        "          # check if file exists with \"/content/drive/MyDrive/LoRA/\" prefix\n",
        "          if not os.path.exists(\"/content/drive/MyDrive/LoRA/\" + drivePath):\n",
        "            print(f\"{drivePath} does not exist. Skipping copying step.\")\n",
        "          else:\n",
        "            drivePath = \"/content/drive/MyDrive/LoRA/\" + drivePath\n",
        "        else:\n",
        "          drivePath = \"/content/drive/MyDrive/\" + drivePath\n",
        "      else:\n",
        "        drivePath = \"/\" + drivePath\n",
        "    !cp {drivePath} {pretrained_model}\n",
        "else:\n",
        "  print(\"\")\n"
      ],
      "metadata": {
        "id": "fWUIPS97FsLL",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title 追加学習モデル LoRA・Textual Inversion･Lycoris（任意）\n",
        "#@markdown [LoRAの使用方法](https://intindex.stars.ne.jp/archives/12296)\n",
        "\n",
        "\n",
        "Lora_name =\"\" #@param [\"\",\"AI_uehara\",\"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",
        "Lora_dict={\n",
        "\t\"AI_uehara\":\"https://civitai.com/api/download/models/23042\",\n",
        "\t\"Anime_Screencap_Style_LoRA\":\"https://civitai.com/api/download/models/5742\" ,\n",
        "\t\"One_Piece_Wano_Style_LoRA\":\"https://civitai.com/api/download/models/6331\" ,\n",
        "\t\"Eula_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Eula/eulaMedium.safetensors\" ,\n",
        "\t\"Yelan_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Yelan/Yelan_Hard.safetensors\" ,\n",
        "\t\"Barbara_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Barbara/Barbara_mediumpruned.safetensors\",\n",
        "\t\"Diluc_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Diluc/Diluc_Mediumpruned.safetensors\" ,\n",
        "\t\"Mona_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Mona/Mona_mediumpruned.safetensors\" ,\n",
        "\t\"Ganyu_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Ganyu/Ganyu_Medium.safetensors\" ,\n",
        "\t\"Jean_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Jean/Jean_Medium.safetensors\" ,\n",
        "\t\"Kujou_Sara_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Kujou%20Sara/kujou%20sara_Heavy.safetensors\",\n",
        "\t\"Lisa_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Lisa/Lisa_Medium.safetensors\",\n",
        "\t\"Raiden_Shogun_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Raiden%20Shogun/raiden%20shogun_LoRA.safetensors\",\n",
        "\t\"Shenhe_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Shenhe/Shenhe_Medium.safetensors\",\n",
        "\t\"Yoimiya_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Yoimiya/Yoimiya_mediumpruned.safetensors\",\n",
        "\t\"Zhongli_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/Zhongli/Zhongli_Medium.safetensors\",\n",
        "\t\"Yae_Miko_LoRA\":\"https://huggingface.co/YoungMasterFromSect/Trauter_LoRAs/resolve/main/LoRA/Genshin-Impact/yae%20miko/yae%20miko_Mediumpruned.safetensors\",\n",
        "}\n",
        "\n",
        "Textual_name =\"\"#@param [\"\",\"Corneos_Makima\",\"Corneo_Power\",\"Yor_Forger\",\"Corneos_POV_Bound_Wrists_Missionary\",\"Corneos_Spitroast_Threesome\",\"Corneos_Anal_sex\",\"Corneos_Side-view_Deepthroat\",\"Corneos_Covering_Breasts-Two_Hands\",\"Corneos_X-shaped_Nipple_Pasties\",\"Corneos_POV_Oral\",\"Corneos_POV_Paizuri\",\"Corneos_Covering_Breasts-One_Arm\",\"Corneos_Arm_Grab_Doggystyle\",\"Corneos_Covering_Breasts-Arms_Crossed\",\"Licking_Oral\"] {allow-input:false}\n",
        "\n",
        "Textual_dict={\n",
        "\t\"Corneos_Makima\":\"https://civitai.com/api/download/models/6197\",\n",
        "\t\"Corneo_Power\":\"https://civitai.com/api/download/models/6027\" ,\n",
        "\t\"Yor_Forger\":\"https://civitai.com/api/download/models/6300\" ,\n",
        "\t\"Corneos_POV_Bound_Wrists_Missionary\":\"https://civitai.com/api/download/models/5405\",\n",
        "\t\"Corneos_Spitroast_Threesome\":\"https://civitai.com/api/download/models/6770\",\n",
        "\t\"Corneos_Anal_sex\":\"https://civitai.com/api/download/models/4597\" ,\n",
        "\t\"Corneos_Side-view_Deepthroat\":\"https://civitai.com/api/download/models/6242\" ,\n",
        "\t\"Corneos_Covering_Breasts-Two_Hands\":\"https://civitai.com/api/download/models/6033\" ,\n",
        "\t\"Corneos_X-shaped_Nipple_Pasties\":\"https://civitai.com/api/download/models/6318\" ,\n",
        "\t\"Corneos_POV_Oral\":\"https://civitai.com/api/download/models/5051\" ,\n",
        "\t\"Corneos_POV_Paizuri\":\"https://civitai.com/api/download/models/5066\",\n",
        "\t\"Corneos_Covering_Breasts-One_Arm\":\"https://civitai.com/api/download/models/6034\",\n",
        "\t\"Corneos_Arm_Grab_Doggystyle\":\"https://civitai.com/api/download/models/5160\" ,\n",
        "\t\"Corneos_Covering_Breasts-Arms_Crossed\":\"https://civitai.com/api/download/models/5746\",\n",
        "\t\"Licking_Oral\":\"https://civitai.com/api/download/models/5746\" ,\n",
        "}\n",
        "\n",
        "#Lycoris-LoRA\n",
        "Lycoris_name =\"\" #@param [\"\",\"準備中\"] {allow-input:false}\n",
        "#準備中\n",
        "Lycoris_dict={\n",
        "\t\t\"\":\"\",\n",
        "}\n",
        "\n",
        "Lora_url = Lora_dict.get(Lora_name)\n",
        "Textual_url = Textual_dict.get(Textual_name)\n",
        "Lycoris_url = Lycoris_dict.get(Lycoris_name)\n",
        "import os\n",
        "\n",
        "def install(Lora_url, Textual_url, Lycoris_url):\n",
        "\n",
        "    if Lora_url:\n",
        "        base_name1 = os.path.basename(Lora_url)\n",
        "        if Lora_url.startswith(\"https://huggingface.co/\"):\n",
        "            !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d /content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora -o {base_name1} {Lora_url}\n",
        "        else:\n",
        "            !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d /content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora {Lora_url}\n",
        "    if Textual_url:\n",
        "        base_name2 = os.path.basename(Textual_url)\n",
        "        !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {Textual_url} -d /content/stable-diffusion-webui/embeddings {Textual_url}\n",
        "    if Lycoris_url:\n",
        "        base_name3 = os.path.basename(Lycoris_url)\n",
        "\n",
        "        if Lycoris_url.startswith(\"https://huggingface.co/\"):\n",
        "            !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d /content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora -o {base_name3} {Lycoris_url}\n",
        "        else:\n",
        "            !aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d /content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora {(Lycoris_url}\n",
        "\n",
        "install(Lora_url, Textual_url, Lycoris_url)\n"
      ],
      "metadata": {
        "id": "QDG0oyNU6R1L",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title  4.WebUIの起動\n",
        "%cd /content/stable-diffusion-webui\n",
        "\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/\\\"sd_model_checkpoint\\\"\\,/\\\"sd_model_checkpoint\\,sd_vae\\,CLIP_stop_at_last_layers\\\"\\,/g' /content/stable-diffusion-webui/modules/shared.py\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",
        "▼より高速な画像生成を行いたい方向け<br>\n",
        "ColabのProに加入するよりも[Paperspace](https://note.com/ai_nios/n/n2203de79bc16)を使用することをおすすめします。<br>\n",
        "<br>"
      ],
      "metadata": {
        "id": "x54Lv69ieDvt"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "#@title　生成した画像フォルダのアップロード/zipファイルダウンロード(任意)\n",
        "import os\n",
        "\n",
        "upload_download = \"Upload_zip_drive and Download_zip\"  # @param [\"Upload_drive\", \"Download_zip\", \"Upload_zip_drive and Download_zip\"]\n",
        "folder_type = \"zip\" #@param [\"raw\",\"zip\"]\n",
        "folder_name = \"generated_image\" # @param {'type': 'string'}\n",
        "\n",
        "if upload_download == \"Upload_drive\":\n",
        "    from google.colab import drive\n",
        "    if not os.path.exists(\"/content/drive\"):\n",
        "        drive.mount(\"/content/drive\")\n",
        "    if folder_type == \"zip\":\n",
        "        !zip -r \"/content/{folder_name}.zip\" \"/content/stable-diffusion-webui/outputs\"\n",
        "        !cp -r \"/content/{folder_name}.zip\" \"/content/drive/MyDrive.zip\"\n",
        "    elif folder_type == \"raw\":\n",
        "        !cp -r \"/content/stable-diffusion-webui/outputs\" \"/content/drive/MyDrive/{folder_name}\"\n",
        "\n",
        "elif upload_download == \"Download_zip\":\n",
        "    from google.colab import files\n",
        "    if folder_type == \"zip\":\n",
        "        !zip -r \"/content/{folder_name}.zip\" \"/content/stable-diffusion-webui/outputs/txt2img-images/\"\n",
        "        files.download(f\"/content/{folder_name}.zip\")\n",
        "    elif folder_type == \"raw\":\n",
        "        print(\"フォルダはzipファイルではないため、ダウンロードできません。\")\n",
        "\n",
        "elif upload_download == \"Upload_zip_drive and Download_zip\":\n",
        "    from google.colab import drive\n",
        "    if not os.path.exists(\"/content/drive\"):\n",
        "        drive.mount(\"/content/drive\")\n",
        "    from google.colab import files\n",
        "    if folder_type == \"zip\":\n",
        "        files.download(f\"/content/{folder_name}.zip\")\n",
        "        !zip -r \"/content/{folder_name}.zip\" \"/content/stable-diffusion-webui/outputs\"\n",
        "        !cp -r \"/content/{folder_name}.zip\" \"/content/drive/MyDrive.zip\"\n",
        "\n",
        "    elif folder_type == \"raw\":\n",
        "        files.download(f\"/content/{folder_name}\")\n",
        "        !cp -r \"/content/stable-diffusion-webui/outputs\" \"/content/drive/MyDrive/{folder_name}\"\n"
      ],
      "metadata": {
        "cellView": "form",
        "id": "NHf0j9NxcjEx"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "\n",
        "▼学習済みモデルの検索<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": [],
      "gpuType": "T4",
      "include_colab_link": true
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}