install_sdxl/yamer_mix_xl_webui_colab.ipynb

{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@markdown ## 구글 드라이브 전체 설치 (최초 한번만 설치)\n",
        "#@markdown <div><font color=\"red\">5GB~7GB정도 필요해요. 구글드라이브 용량을 확인해주세요</font></div>  \n",
        "#@markdown * 설치후에 <a href=\"https://github.com/ninjaneural/webui/blob/master/memo/install.md\">이곳</a>에서 실행방법을 확인해주세요*\n",
        "Google_Install = True\n",
        "\n",
        "#@markdown **WebUI Version**\n",
        "TagVersion = 'v1.6.0' #@param [\"v1.3.2\", \"v1.4.1\", \"v1.5.2\", \"v1.6.0\"]\n",
        "\n",
        "#@markdown **폴더 설명** \n",
        "#@markdown * install/outputs : 생성된 이미지들이 저장되요\n",
        "#@markdown * install/models/Stable-diffusion/ : 모델(checkpoint)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * install/models/Lora : 로라(LoRA)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * install/models/LyCORIS : 라이코리스(LyCORIS)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * install/embeddings : 임베딩(Textual Inversion)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * install/hypernetworks : 하이퍼워크(Hyperworks)를 넣어주면 읽어올수 있어요\n",
        "\n",
        "#@markdown **초기모델(checkpoint) URL**\n",
        "Checkpoint_Url = 'https://civitai.com/api/download/models/148515?type=Model&format=SafeTensor&size=full&fp=fp16' #@param {type:\"string\"}\n",
        "#@markdown **초기모델 파일명**\n",
        "Checkpoint_Filename = 'yamer_mix_xl.safetensors' #@param {type:\"string\"}\n",
        "\n",
        "#@markdown -----\n",
        "#@markdown *추가익스텐션*\n",
        "\n",
        "ControlNet = True \n",
        "ADetailer = True\n",
        "\n",
        "#@markdown **Dectect Detailer(ddetailer)를 사용하지 않으면 체크를 해제해주세요**\n",
        "DDetailer = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown **Segment Anything을 사용하지 않으면 체크를 해제해주세요**\n",
        "SegmentAnything = False #@param {type:\"boolean\"}\n",
        "\n",
        "NotebookVersion = 'install'\n",
        "Workspace = 'drive/MyDrive/install'\n",
        "\n",
        "if Google_Install:\n",
        "  from google.colab import drive\n",
        "  drive.mount('/content/drive')\n",
        "\n",
        "if Google_Install:\n",
        "  !mkdir -p /content/{Workspace}\n",
        "\n",
        "Checkpoint_SavePath = f'/content/{Workspace}/models/Stable-diffusion'\n",
        "\n",
        "!apt -y install -qq aria2\n",
        "\n",
        "%cd /content\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/install_{NotebookVersion}.sh -O /content/install.sh\n",
        "!bash /content/install.sh {Workspace} {TagVersion} {ControlNet} {DDetailer} {ADetailer} {SegmentAnything}\n",
        "\n",
        "%cd /content/{Workspace}\n",
        "\n",
        "# checkpoint select\n",
        "!sed -i -e 's/\"sd_model_checkpoint\": \"\",/\"sd_model_checkpoint\": \"{Checkpoint_Filename}\",/g' /content/{Workspace}/config.json\n",
        "\n",
        "# upscaler\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/{Workspace}/models/ESRGAN -o 4x-UltraSharp.pth\n",
        "\n",
        "# checkpoint\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"{Checkpoint_Url}\" -d {Checkpoint_SavePath} -o {Checkpoint_Filename}\n",
        "\n",
        "# vae\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt -d {Checkpoint_SavePath} -o vae-ft-mse-840000-ema-pruned.vae.pt\n",
        "\n",
        "# lora\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://civitai.com/api/download/models/62833?type=Model&format=SafeTensor\" -d /content/{Workspace}/models/Lora -o add_detail.safetensors\n",
        "\n",
        "# embeddings\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.pt -d /content/{Workspace}/embeddings -o EasyNegative.pt\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/AsciiP/badhandv4/resolve/main/badhandv4.pt -d /content/{Workspace}/embeddings -o badhandv4.pt\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://civitai.com/api/download/models/60938?type=Negative&format=Other\" -d /content/{Workspace}/embeddings/ -o negative_hand.pt\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://civitai.com/api/download/models/60095?type=Negative&format=Other\" -d /content/{Workspace}/embeddings/ -o bad_prompt_version2.pt\n",
        "\n",
        "!git clone https://github.com/salesforce/BLIP.git /content/drive/MyDrive/install/repositories/BLIP\n",
        "%cd /content/drive/MyDrive/install/repositories/BLIP\n",
        "!git reset --hard 48211a1594f1321b00f14c9f7a5b4813144b2fb9\n",
        "\n",
        "!git clone https://github.com/sczhou/CodeFormer.git /content/drive/MyDrive/install/repositories/CodeFormer\n",
        "%cd /content/drive/MyDrive/install/repositories/CodeFormer\n",
        "!git reset --hard c5b4593074ba6214284d6acd5f1719b6c5d739af\n",
        "\n",
        "!git clone https://github.com/Stability-AI/generative-models.git /content/drive/MyDrive/install/repositories/generative-models\n",
        "%cd /content/drive/MyDrive/install/repositories/generative-models\n",
        "!git reset --hard 45c443b316737a4ab6e40413d7794a7f5657c19f\n",
        "\n",
        "!git clone https://github.com/crowsonkb/k-diffusion.git /content/drive/MyDrive/install/repositories/k-diffusion\n",
        "%cd /content/drive/MyDrive/install/repositories/k-diffusion\n",
        "!git reset --hard ab527a9a6d347f364e3d185ba6d714e22d80cb3c\n",
        "\n",
        "!git clone https://github.com/Stability-AI/stablediffusion.git /content/drive/MyDrive/install/repositories/stable-diffusion-stability-ai\n",
        "%cd /content/drive/MyDrive/install/repositories/stable-diffusion-stability-ai\n",
        "!git reset --hard cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf\n",
        "\n",
        "print(\"------------------------------------------------------------------------------------------------\")\n",
        "print(\"설치완료!\")\n",
        "print(\"https://github.com/ninjaneural/webui/blob/master/memo/install.md 이 페이지를 참고해서 실행해주세요\")\n"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "collapsed_sections": [],
      "private_outputs": true,
      "provenance": []
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.10.11"
    },
    "vscode": {
      "interpreter": {
        "hash": "32c23ec9c124fdbef417fa7c12f036622d7e14abb755fc2527a38410604ea1a4"
      }
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}