install/beautyfool_colab.ipynb

{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "from IPython.utils import capture\n",
        "from IPython.display import clear_output\n",
        "\n",
        "#@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.7.0' #@param [\"v1.3.2\", \"v1.4.1\", \"v1.5.2\", \"v1.6.0\", \"v1.7.0\", \"v1.8.0\"]\n",
        "\n",
        "#@markdown **초기모델(checkpoint) URL**\n",
        "Checkpoint_Url = 'https://civitai.com/api/download/models/215628' #@param {type:\"string\"}\n",
        "#@markdown **초기모델 파일명**\n",
        "Checkpoint_Filename = 'beautyfool_v30.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",
        "#@markdown **AnimateDiff 사용 (WebUI 버전 v1.5.2이하에서는 동작하지 않을수 있어요)**\n",
        "AnimateDiff = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown **mov2mov 사용**\n",
        "Mov2mov = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown **Deforum 사용**\n",
        "Deforum = 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",
        "%cd /content\n",
        "\n",
        "print('install...')\n",
        "with capture.capture_output() as cap:\n",
        "  !apt -y install -qq aria2\n",
        "  !wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/install_{NotebookVersion}.sh -O /content/install.sh\n",
        "  !chmod 755 /content/install.sh\n",
        "  !./install.sh {Workspace} {TagVersion} {ControlNet} {DDetailer} {ADetailer} {SegmentAnything} {AnimateDiff} {Mov2mov} {Deforum}\n",
        "  !rm install.sh\n",
        "  %cd /content/{Workspace}\n",
        "  !wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/config.json -O ./config.json\n",
        "  !wget https://huggingface.co/ninjaneural/webui/resolve/main/direct/{TagVersion}/repositories.tar\n",
        "  !tar -xvf repositories.tar\n",
        "  !rm repositories.tar\n",
        "\n",
        "%cd /content/{Workspace}\n",
        "\n",
        "# 추가 익스텐션\n",
        "# git clone https://github.com/vladmandic/sd-extension-system-info ./extensions/sd-extension-system-info\n",
        "\n",
        "# checkpoint select\n",
        "!sed -i -e 's/\"sd_model_checkpoint\": \"\",/\"sd_model_checkpoint\": \"{Checkpoint_Filename}\",/g' /content/{Workspace}/config.json\n",
        "\n",
        "print('download...')\n",
        "with capture.capture_output() as cap:\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",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://huggingface.co/latent-consistency/lcm-lora-sdv1-5/resolve/main/pytorch_lora_weights.safetensors\" -d /content/{Workspace}/models/Lora -o lcm-lora-sd15.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",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://civitai.com/api/download/models/5637?type=Model&format=PickleTensor&size=full&fp=fp16\" -d /content/{Workspace}/embeddings/ -o ng_deepnegative_v1_75t.pt\n",
        "\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/direct/{TagVersion}/directui.py -O /content/{Workspace}/directui.py\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/direct/{TagVersion}/launch_utils.py -O /content/{Workspace}/modules/launch_utils.py\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
}