sdxl_0_9.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyNZzn7wWLlxQynTk+fBEtGu",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/detektor777/colab_list_image/blob/main/sdxl_0_9.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "CrRa7ZVpaPR5"
      },
      "outputs": [],
      "source": [
        "#@title ##**Install** { display-mode: \"form\" }\n",
        "%%capture\n",
        "from IPython.display import clear_output\n",
        "from pathlib import Path\n",
        "\n",
        "!npm install -g localtunnel > /dev/null 2>&1\n",
        "OPTIONS = {}\n",
        "\n",
        "USE_GOOGLE_DRIVE = False\n",
        "UPDATE_COMFY_UI = False\n",
        "WORKSPACE = 'ComfyUI'\n",
        "OPTIONS['USE_GOOGLE_DRIVE'] = USE_GOOGLE_DRIVE\n",
        "OPTIONS['UPDATE_COMFY_UI'] = UPDATE_COMFY_UI\n",
        "\n",
        "if OPTIONS['USE_GOOGLE_DRIVE']:\n",
        "    !echo \"Mounting Google Drive...\"\n",
        "    %cd /\n",
        "\n",
        "    from google.colab import drive\n",
        "    drive.mount('/content/drive')\n",
        "\n",
        "    WORKSPACE = \"/content/drive/MyDrive/ComfyUI\"\n",
        "    %cd /content/drive/MyDrive\n",
        "\n",
        "![ ! -d $WORKSPACE ] && echo -= Initial setup ComfyUI =- && git clone https://github.com/comfyanonymous/ComfyUI\n",
        "%cd $WORKSPACE\n",
        "\n",
        "if OPTIONS['UPDATE_COMFY_UI']:\n",
        "  !echo -= Updating ComfyUI =-\n",
        "  !git pull\n",
        "\n",
        "!echo -= Install dependencies =-\n",
        "!pip install xformers!=0.0.18 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118 --extra-index-url https://download.pytorch.org/whl/cu117\n",
        "\n",
        "clear_output()\n",
        "\n",
        "#2\n",
        "\n",
        "MODEL_LINK_1 = \"https://huggingface.co/modelslover/notsdxl/resolve/main/sd_xl_base_0.9%20(3).safetensors\"\n",
        "MODEL_SAFETENSORS_1 = True\n",
        "MODEL_LINK_2 = \"https://huggingface.co/modelslover/notsdxl/resolve/main/diffusion_pytorch_model.fp16.safetensors\"\n",
        "MODEL_SAFETENSORS_2 = True\n",
        "\n",
        "MODEL_LINKS = [\n",
        "    {\n",
        "        'link': MODEL_LINK_1,\n",
        "        'safetensors': MODEL_SAFETENSORS_1\n",
        "    },\n",
        "    {\n",
        "        'link': MODEL_LINK_2,\n",
        "        'safetensors': MODEL_SAFETENSORS_2\n",
        "    },\n",
        "\n",
        "]\n",
        "\n",
        "if all(model['link'] for model in MODEL_LINKS):\n",
        "  pth = '/content/ComfyUI/models/checkpoints/'\n",
        "  for i, model in enumerate(MODEL_LINKS):\n",
        "    link = model['link']\n",
        "    safetensors = model['safetensors']\n",
        "\n",
        "    if safetensors:\n",
        "        modelname = \"model.safetensors\"\n",
        "    else:\n",
        "        modelname = \"model.ckpt\"\n",
        "\n",
        "    dwnld = pth + f\"model{i+1}.{modelname.split('.')[1]}\"\n",
        "    print(f'\u001b[1;32mDownloading model {i+1}...')\n",
        "    !gdown --fuzzy -O $dwnld \"$link\"\n",
        "    print(f'\u001b[1;32mModel {i+1} downloaded!')\n",
        "\n",
        "  print('\u001b[1;32mAll models downloaded successfully!')\n",
        "else:\n",
        "  print('\u001b[1;31mPlease provide links and corresponding safetensors parameters for all models!')\n"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@title ##**Open** { display-mode: \"form\" }\n",
        "\n",
        "!fuser -k 8188/tcp\n",
        "import subprocess\n",
        "import threading\n",
        "import time\n",
        "import socket\n",
        "import urllib.request\n",
        "from IPython.display import HTML, display\n",
        "\n",
        "button_displayed = False\n",
        "\n",
        "def iframe_thread(port):\n",
        "    global button_displayed\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",
        "\n",
        "    if not button_displayed:\n",
        "        password = urllib.request.urlopen('https://ipv4.icanhazip.com').read().decode('utf8').strip(\"\\n\")\n",
        "\n",
        "        p = subprocess.Popen([\"lt\", \"--port\", \"{}\".format(port)], stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)\n",
        "        for line in p.stdout:\n",
        "            if line.decode().startswith('your url is:'):\n",
        "                url = line.decode().split(': ')[1].strip()\n",
        "                break\n",
        "\n",
        "        # Создаем HTML с кнопками\n",
        "        html = f\"\"\"\n",
        "        <script>\n",
        "        function copyPassword() {{\n",
        "            navigator.clipboard.writeText(\"{password}\");\n",
        "            alert(\"Password copied to clipboard!\");\n",
        "        }}\n",
        "        </script>\n",
        "        <button onclick=\"window.open('{url}', '_blank')\">Open Tunnel</button>\n",
        "        <button onclick=\"copyPassword()\">Copy Password</button>\n",
        "        <p>Password: {password}</p>\n",
        "        \"\"\"\n",
        "        display(HTML(html))\n",
        "        button_displayed = True\n",
        "\n",
        "if not button_displayed:\n",
        "    threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n",
        "\n",
        "!python main.py --dont-print-server > /dev/null 2>&1"
      ],
      "metadata": {
        "id": "P94PaMEFaize"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}