deforum/yabal_mix_true25d_webui_colab.ipynb

{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@markdown ## 구글드라이브 연동\n",
        "#@markdown **생성한 이미지를 구글드라이브에 자동저장하려면 체크해주세요**\n",
        "Google_Drive = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown **초기모델(checkpoint)을 구글드라이브에 저장하려면 체크해주세요**\n",
        "#@markdown <div><font color=\"red\">모델 파일사이즈(2GB ~ 7GB)가 크니 구글드라이브 용량을 확인해주세요</div>\n",
        "#@markdown <div><font color=\"red\">대신 한번받으면 다음 실행부터는 빨라져요</div>\n",
        "Checkpoint_Google_Save = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown -----\n",
        "#@markdown *터널링*\n",
        "\n",
        "#@markdown **Ngrok**\n",
        "Ngrok_Key = '' #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **Localtunnel**\n",
        "Localtunnel = False #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown -----\n",
        "#@markdown *아래는 설정할 필요 없어요*\n",
        "\n",
        "#@markdown **구글드라이브에 연결 폴더**\n",
        "Google_Drive_Dir = 'webui' #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **폴더 설명** : 미리 만드셔도 되고 없으면 자동으로 생성해요\n",
        "#@markdown * webui/output : 생성된 이미지들이 저장되요\n",
        "#@markdown * webui/checkpoint : 모델(checkpoint)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * webui/lora : 로라(LoRA)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * webui/embedding : 임베딩(Textual Inversion)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * webui/hyperwork : 하이퍼워크(Hyperworks)를 넣어주면 읽어올수 있어요\n",
        "#@markdown * webui/wildcards : 와일드카드(Wildcards)를 넣어주면 읽어올수 있어요\n",
        "\n",
        "#@markdown **초기모델(checkpoint) URL**\n",
        "Checkpoint_Url = 'https://huggingface.co/digiplay/YabaLMixTrue25D_V2.0/resolve/main/yabalMixTrue25dV20_ver20.safetensors?download=true' #@param {type:\"string\"}\n",
        "#@markdown **초기모델 파일명**\n",
        "Checkpoint_Filename = 'yabaLMix_true25d_v5.safetensors' #@param {type:\"string\"}\n",
        "\n",
        "#@markdown -----\n",
        "#@markdown *추가익스텐션*\n",
        "\n",
        "#@markdown **ControlNet을 사용하지 않으면 체크를 해제해주세요**\n",
        "ControlNet = True #@param {type:\"boolean\"}\n",
        "Workspace = 'ui'\n",
        "NotebookVersion = 'deforum'\n",
        "\n",
        "if Google_Drive:\n",
        "  from google.colab import drive\n",
        "  drive.mount('/content/drive')\n",
        "\n",
        "Checkpoint_SavePath = f'/content/{Workspace}/models/Stable-diffusion'\n",
        "if Google_Drive and Checkpoint_Google_Save:\n",
        "  Checkpoint_SavePath = f'/content/{Workspace}/models/Stable-diffusion/google'\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} {ControlNet}\n",
        "\n",
        "# 구글드라이브 연결\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/link_google_drive.sh -O /content/link_google_drive.sh\n",
        "!bash /content/link_google_drive.sh {Workspace} {Google_Drive} {Google_Drive_Dir}\n",
        "\n",
        "# 커스터마이징\n",
        "%cd /content/{Workspace}\n",
        "!bash /content/drive/MyDrive/{Google_Drive_Dir}/{NotebookVersion}/install.sh {Workspace}\n",
        "if Google_Drive:\n",
        "  !cp -f /content/drive/MyDrive/{Google_Drive_Dir}/{NotebookVersion}/config.json /content/{Workspace}/config.json\n",
        "  !cp -f /content/drive/MyDrive/{Google_Drive_Dir}/{NotebookVersion}/ui-config.json /content/{Workspace}/ui-config.json\n",
        "  !cp -f /content/drive/MyDrive/{Google_Drive_Dir}/{NotebookVersion}/styles.csv /content/{Workspace}/styles.csv\n",
        "\n",
        "# checkpoint select\n",
        "!sed -i -e 's/\"sd_model_checkpoint\": \"\",/\"sd_model_checkpoint\": \"{Checkpoint_Filename}\",/g' /content/{Workspace}/config.json\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",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt -d /content/{Workspace}/models/VAE -o kl-f8-anime2.ckpt\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://civitai.com/api/download/models/63006?type=Model&format=SafeTensor\" -d /content/{Workspace}/models/Lora -o LowRA.safetensors\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \"https://civitai.com/api/download/models/32988?type=Model&format=SafeTensor&size=full&fp=fp16\" -d /content/{Workspace}/models/Lora -o blindbox_V1Mix.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://huggingface.co/yesyeahvh/bad-hands-5/resolve/main/bad-hands-5.pt -d /content/{Workspace}/embeddings -o bad-hands-5.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",
        "if Localtunnel:\n",
        "  !npm install -g localtunnel\n",
        "\n",
        "  import subprocess\n",
        "  import threading\n",
        "  import time\n",
        "  import socket\n",
        "  import urllib.request\n",
        "\n",
        "  def iframe_thread(port):\n",
        "    while True:\n",
        "        time.sleep(0.5)\n",
        "        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)\n",
        "        result = sock.connect_ex(('127.0.0.1', port))\n",
        "        if result == 0:\n",
        "          break\n",
        "        sock.close()\n",
        "    print(\"\\nComfyUI finished loading, trying to launch localtunnel (if it gets stuck here localtunnel is having issues)\\n\")\n",
        "\n",
        "    print(\"The password/enpoint ip for localtunnel is:\", urllib.request.urlopen('https://ipv4.icanhazip.com').read().decode('utf8').strip(\"\\n\"))\n",
        "    p = subprocess.Popen([\"lt\", \"--port\", \"{}\".format(port)], stdout=subprocess.PIPE)\n",
        "    for line in p.stdout:\n",
        "      print(line.decode(), end='')\n",
        "\n",
        "\n",
        "  threading.Thread(target=iframe_thread, daemon=True, args=(7860,)).start()\n",
        "\n",
        "if Ngrok_Key:\n",
        "  !python launch.py --xformers --no-half-vae --theme dark --ngrok {Ngrok_Key}\n",
        "else:\n",
        "  !python launch.py --share --xformers --no-half-vae --theme dark --gradio-queue\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
}