comfyui/dreamshaper_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 **구글드라이브에 연결 폴더**\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/lycoris : 라이코리스(LyCORIS)를 넣어주면 읽어올수 있어요\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://civitai.com/api/download/models/109123?type=Model&format=SafeTensor&size=pruned&fp=fp16' #@param {type:\"string\"}\n",
        "#@markdown **초기모델 파일명**\n",
        "Checkpoint_Filename = 'DreamShaper_7_BakedVae_pruned.safetensors' #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **ControlNet을 사용하지 않으면 체크를 해제해주세요**\n",
        "ControlNet = True #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown **ComfyUI Impact Pack**\n",
        "ComfyUI_Impact_Pack = False  #@param {type:\"boolean\"}\n",
        "\n",
        "Workspace = 'ui'\n",
        "\n",
        "if Google_Drive:\n",
        "  from google.colab import drive\n",
        "  drive.mount('/content/drive')\n",
        "\n",
        "Checkpoint_SavePath = f'/content/{Workspace}/models/checkpoints'\n",
        "if Google_Drive and Checkpoint_Google_Save:\n",
        "  Checkpoint_SavePath = f'/content/{Workspace}/models/checkpoints/google'\n",
        "\n",
        "!apt -y install -qq aria2\n",
        "\n",
        "%cd /content\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/install_comfyui.sh -O /content/install.sh\n",
        "!bash /content/install.sh {Workspace} {ComfyUI_Impact_Pack} {ControlNet}\n",
        "\n",
        "# 구글드라이브 연결\n",
        "!wget https://raw.githubusercontent.com/neuralninja22/colab/master/misc/link_google_drive_comfyui.sh -O /content/link_google_drive_comfyui.sh\n",
        "!bash /content/link_google_drive_comfyui.sh {Workspace} {Google_Drive} {Google_Drive_Dir}\n",
        "\n",
        "%cd /content/{Workspace}\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",
        "# upscaler\n",
        "!wget -c https://huggingface.co/sberbank-ai/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth -P /content/{Workspace}/models/upscale_models/\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/upscale_models -o 4x-UltraSharp.pth\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 /content/{Workspace}/models/vae -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/loras -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/loras -o LowRA.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}/models/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}/models/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}/models/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}/models/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}/models/embeddings -o bad_prompt_version2.pt\n",
        "\n",
        "# Run ComfyUI with localtunnel (Recommended Way)\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=(8188,)).start()\n",
        "\n",
        "!python main.py --dont-print-server"
      ]
    }
  ],
  "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
}