import gradio as gr def greet(name): return "Hello " + name + "!" demo = gr.Interface(fn=greet, inputs="text", outputs="text") { "cells": [ { "cell_type": "markdown", "metadata": { "id": "SKi-oMBwhutR" }, "source": [ "# AUTOMATIC1111 Stable Diffusion WebUI 1.5 + ChilloutMix Checkpoint\n", "\n", "For generating AI Onlyfans\n", "\n", "> [@nang](https://github.com/nathan-149) \n", "> Based off of: [@wibus-wee](https://github.com/wibus-wee)\n", "> Reference: [camenduru/stable-diffusion-webui-colab](https://github.com/camenduru/stable-diffusion-webui-colab)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "oL8eHMWfLzO7" }, "outputs": [], "source": [ "#@title 2. Check GPU & Dev Environment\n", "\n", "from IPython.display import display\n", "import ipywidgets as widgets\n", "import requests \n", "\n", "import os, subprocess\n", "paperspace_m4000 = False\n", "#@markdown Paperspace platform?\n", "isPaperspace = False #@param {type:\"boolean\"}\n", "try:\n", " subprocess.run(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'], stdout=subprocess.PIPE)\n", " if 'M4000' in subprocess.run(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'], stdout=subprocess.PIPE).stdout.decode('utf-8'):\n", " print(\"WARNING: You're using Quadro M4000 GPU，xformers won't work。\")\n", " paperspace_m4000 = True\n", " isPaperspace = True\n", " else:\n", " print(\"Your GPU is suitable - \" + subprocess.run(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'], stdout=subprocess.PIPE).stdout.decode('utf-8') + \"。\")\n", " print(\"Platform: Paperspace\" if isPaperspace else \"Platform: Colab\")\n", "except:\n", " print(\"No GPU appears to be available. Please check your runtime type\")\n", " exit()\n", "\n", "rootDir = isPaperspace and '/tmp' or '/content'\n", "stableDiffusionWebUIInstalled = os.path.exists(rootDir + '/stable-diffusion-webui')\n", "%store rootDir\n", "%store paperspace_m4000 \n", "%store isPaperspace\n", "%store stableDiffusionWebUIInstalled" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "SaAJk33ppFw1" }, "outputs": [], "source": [ "import requests\n", "#@title 3. Install dependencies and extensions \n", "\n", "#@markdown ## **Extensions**\n", "\n", "#@markdown xformer\n", "xformersInstall = True #@param {type:\"boolean\"}\n", "#@markdown ControlNet\n", "controlNetExtension = False #@param {type:\"boolean\"}\n", "#@markdown OpenPose Editor\n", "openPoseExtension = False #@param {type:\"boolean\"}\n", "#@markdown Civitai Browser\n", "civitaiBrowserExtension = False #@param {type:\"boolean\"}\n", "#@markdown HuggingFace \n", "huggingFaceExtension = False #@param {type:\"boolean\"}\n", "#@markdown Images Browser \n", "imagesBrowserExtension = False #@param {type:\"boolean\"}\n", "#@markdown Additional Networks \n", "additionalNetworksExtension = True #@param {type:\"boolean\"}\n", "#@markdown Deforum \n", "deforumExtension = False #@param {type:\"boolean\"}\n", "#@markdown Kohya sd-scripts \n", "kohyaExtension = False #@param {type:\"boolean\"}\n", "#@markdown DreamBooth \n", "dreamBoothExtension = False #@param {type:\"boolean\"}\n", "\n", "\n", "#@markdown ---\n", "#@markdown ## **Textual Inversion**\n", "#@markdown Ulzzang-6500 (Korean doll aesthetic)\n", "ulzzang6500 = True #@param {type:\"boolean\"}\n", "#@markdown Pure Eros Face\n", "pureErosFace = True #@param {type:\"boolean\"}\n", "\n", "#@markdown ---\n", "\n", "#@markdown ## **Startup Options**\n", "\n", "#@markdown API Support\n", "apiSupport = True #@param {type:\"boolean\"}\n", "\n", "textualInversionDownloadIDs = {\n", " 'ulzzang6500': 8109,\n", " 'pureErosFace': 4514,\n", "}\n", "\n", "def getLatestModelDownloadURL(id):\n", " try:\n", " if type(id) == int:\n", " res = requests.get(endpoint + '/' + str(id)).json()\n", " latest = res['modelVersions'][0]\n", " downloadLink = latest['files'][0]['downloadUrl']\n", " name = latest['files'][0]['name']\n", " return {\n", " 'url': downloadLink,\n", " 'name': name\n", " }\n", " else:\n", " return {\n", " 'url': id,\n", " 'name': id.split('/')[-1]\n", " }\n", " except:\n", " print(\"Lora model \" + str(id) + \" not found. Skip.\")\n", " return None\n", "\n", "def getSpecificModelDownloadURL(id, version):\n", " try:\n", " if type(id) == int:\n", " res = requests.get(endpoint + '/' + str(id)).json()\n", " for modelVersion in res['modelVersions']:\n", " if modelVersion['name'] == version:\n", " # if modelVersion[\"baseModel\"] != \"SD 1.5\":\n", " # print(\"Lora model \" + str(id) + \" is not SD 1.5, may not work. Skip.\")\n", " # return None\n", " downloadLink = modelVersion['files'][0]['downloadUrl']\n", " name = modelVersion['files'][0]['name']\n", " return {\n", " 'url': downloadLink,\n", " 'name': name\n", " }\n", " else:\n", " return {\n", " 'url': id,\n", " 'name': id.split('/')[-1]\n", " }\n", " except:\n", " print(\"Lora model \" + str(id) + \" version \" + version + \" not found. Skip.\")\n", " return None\n", "\n", "def getTextualInversionDownloadURLs():\n", " downloadURLs = []\n", " for key in textualInversionDownloadIDs:\n", " if not eval(key): # skip if not selected\n", " continue\n", " if type(textualInversionDownloadIDs[key]) is int:\n", " downloadURLs.append(getLatestModelDownloadURL(textualInversionDownloadIDs[key]))\n", " elif type(textualInversionDownloadIDs[key]) is dict: # {'id': 123, 'version': 'v1.0'}\n", " downloadURLs.append(getSpecificModelDownloadURL(textualInversionDownloadIDs[key]['id'], textualInversionDownloadIDs[key]['version']))\n", " elif type(textualInversionDownloadIDs[key]) is str: # url\n", " downloadURLs.append({ 'url': textualInversionDownloadIDs[key], 'name': textualInversionDownloadIDs[key].split('/')[-1] })\n", " downloadURLs = [x for x in downloadURLs if x is not None]\n", " return downloadURLs\n", " \n", "textualInversionDownloadURLs = getTextualInversionDownloadURLs()\n", "\n", "%store -r paperspace_m4000 \n", "%store -r isPaperspace\n", "%store -r rootDir \n", "%store -r checkpoints\n", "%store -r downloadLinks\n", "%store -r stableDiffusionWebUIInstalled\n", "\n", "import subprocess\n", "\n", "!apt-get -y install -qq aria2\n", "ariaInstalled = False\n", "\n", "try:\n", " subprocess.run(['aria2c', '--version'], stdout=subprocess.PIPE)\n", " ariaInstalled = True\n", "except:\n", " pass\n", "\n", "if paperspace_m4000:\n", " if xformersInstall:\n", " !pip install ninja\n", " !pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers\n", " !pip install -q --pre triton\n", "else:\n", " !pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl\n", " !pip install -q --pre triton\n", " \n", "\n", "!git clone -b v2.0 https://github.com/nathan-149/stable-diffusion-webui {rootDir}/stable-diffusion-webui\n", "!wget https://raw.githubusercontent.com/nathan-149/stable-diffusion-webui-scripts/main/run_n_times.py -O {rootDir}/stable-diffusion-webui/scripts/run_n_times.py\n", "if deforumExtension:\n", " !git clone https://github.com/deforum-art/deforum-for-automatic1111-webui {rootDir}/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui\n", "if imagesBrowserExtension:\n", " !git clone https://github.com/AlUlkesh/stable-diffusion-webui-images-browser {rootDir}/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser\n", "if huggingFaceExtension:\n", " !git clone https://github.com/camenduru/stable-diffusion-webui-huggingface {rootDir}/stable-diffusion-webui/extensions/stable-diffusion-webui-huggingface\n", "if civitaiBrowserExtension:\n", " !git clone -b v2.0 https://github.com/camenduru/sd-civitai-browser {rootDir}/stable-diffusion-webui/extensions/sd-civitai-browser\n", "if openPoseExtension:\n", " !git clone https://github.com/camenduru/openpose-editor {rootDir}/stable-diffusion-webui/extensions/openpose-editor\n", "if controlNetExtension:\n", " !git clone https://github.com/Mikubill/sd-webui-controlnet {rootDir}/stable-diffusion-webui/extensions/sd-webui-controlnet\n", "if additionalNetworksExtension:\n", " !git clone https://github.com/kohya-ss/sd-webui-additional-networks {rootDir}/stable-diffusion-webui/extensions/sd-webui-additional-networks\n", "if kohyaExtension:\n", " !git clone https://github.com/ddpn08/kohya-sd-scripts-webui.git {rootDir}/stable-diffusion-webui/extensions/kohya-sd-scripts-webui\n", "if dreamBoothExtension:\n", " !git clone https://github.com/d8ahazard/sd_dreambooth_extension {rootDir}/stable-diffusion-webui/extensions/sd_dreambooth_extension\n", "\n", "if isPaperspace:\n", " %cd /stable-diffusion-webui\n", "else:\n", " %cd {rootDir}/stable-diffusion-webui\n", "\n", "\n", "webuiControlNetModels = [\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_canny-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_depth-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_hed-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_mlsd-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_normal-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_openpose-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_scribble-fp16.safetensors\",\n", " \"https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_seg-fp16.safetensors\",\n", "]\n", "annotatorLink = [\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/hand_pose_model.pth\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/body_pose_model.pth\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/dpt_hybrid-midas-501f0c75.pt\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/mlsd_large_512_fp32.pth\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/mlsd_tiny_512_fp32.pth\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/network-bsds500.pth\",\n", " \"https://huggingface.co/ckpt/ControlNet/resolve/main/upernet_global_small.pth\",\n", "]\n", "\n", "def ariaDownload(downloadLink, checkpoint, path):\n", " if (type(downloadLink) == list and type(checkpoint) == list):\n", " for i in downloadLink:\n", " !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {i} -d {path} -o {checkpoint[downloadLink.index(i)]}\n", " else:\n", " !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {downloadLink} -d {path} -o {checkpoint}\n", "def wgetDownload(downloadLink, checkpoint, path):\n", " if (type(downloadLink) == list and type(checkpoint) == list):\n", " for i in downloadLink:\n", " !wget -c {i} -P {path} -O {checkpoint[downloadLink.index(i)]}\n", " else:\n", " !wget -c {downloadLink} -P {path} -O {checkpoint}\n", "def autoDetectDownload(downloadLink, checkpoint, path):\n", " if ariaInstalled:\n", " ariaDownload(downloadLink, checkpoint, path)\n", " else:\n", " wgetDownload(downloadLink, checkpoint, path)\n", "\n", "if controlNetExtension:\n", " for model in webuiControlNetModels:\n", " autoDetectDownload(model, model.split('/')[-1], rootDir + \"/stable-diffusion-webui/extensions/sd-webui-controlnet/models\")\n", " for model in annotatorLink:\n", " autoDetectDownload(model, model.split('/')[-1], rootDir + \"/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator\")\n", "for model in textualInversionDownloadURLs:\n", " autoDetectDownload(model[\"url\"], model[\"name\"], rootDir + \"/stable-diffusion-webui/embeddings\")\n", "\n", "if additionalNetworksExtension:\n", " !ln -s {rootDir}/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora {rootDir}/stable-diffusion-webui/models/Lora\n", "\n", "\n", "stableDiffusionWebUIInstalled = True\n", "%store stableDiffusionWebUIInstalled\n", "\n", "%cd {rootDir}/stable-diffusion-webui\n", "!sed -i -e '''/prepare_environment()/a\\ os.system\\(f\\\"\"\"sed -i -e ''\\\"s/self.logvar\\\\[t\\\\]/self.logvar\\\\[t.item()\\\\]/g\\\"'' {rootDir}/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/models/diffusion/ddpm.py\"\"\")''' {rootDir}/stable-diffusion-webui/launch.py\n", "!sed -i -e '''/prepare_environment()/a\\ os.system\\(f\\\"\"\"sed -i -e ''\\\"s/dict()))/dict())).cuda()/g\\\"'' {rootDir}/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py\"\"\")''' {rootDir}/stable-diffusion-webui/launch.py\n", "if dreamBoothExtension:\n", " !export REQS_FILE=\"./extensions/sd_dreambooth_extension/requirements.txt\"\n", "\n", "\n" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "ht3hxbzPFL3G" }, "outputs": [], "source": [ "#@title Download Chilloutmix Checkpoint\n", "\n", "checkpoint = 'chilloutmix.safetensors' #@param [\"chilloutmix.safetensors\"]\n", "\n", "downloadLink = 'https://civitai.com/api/download/models/11745' #@param \n", "\n", "\n", "!wget -c {downloadLink} -O /content/stable-diffusion-webui/models/Stable-diffusion/{checkpoint}\n", "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd14/resolve/main/sd-v1-4.ckpt -d /content/stable-diffusion-webui/models/Stable-diffusion -o sd-v1-4.ckpt\n", "\n" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "D7tw6cMcMOba" }, "outputs": [], "source": [ "#@title Download Loras\n", "\n", "loraLinks = dict((\n", " ('koreanDollLikeness_v15.safetensors', 'https://civitai.com/api/download/models/31284'),\n", " ('xswltry1.safetensors', 'https://civitai.com/api/download/models/29131'),\n", " ('liyuuLora_liyuuV1.safetensors', 'https://civitai.com/models/9997/liyuu-lora'),\n", " ('aiBeautyIthlinni_ithlinniV1.safetensors', 'https://civitai.com/api/download/models/19671'),\n", " ('Cute_girl_mix4.safetensors', 'https://civitai.com/api/download/models/16677'),\n", " ('breastinclassBetter_v141.safetensors', 'https://civitai.com/api/download/models/23250'),\n", " ('chilloutmixss_xss10.safetensors', 'https://huggingface.co/HankChang/chilloutmixss_xss10/resolve/main/chilloutmixss_xss10.safetensors'),\n", " ('moxin.safetensors', 'https://civitai.com/api/download/models/14856'),\n", " ('legspread10.safetensors', 'https://civitai.com/api/download/models/29760'),\n", " ('taiwan.safetensors', 'https://civitai.com/api/download/models/20684')\n", "))\n", "\n", "\n", "for lora, link in loraLinks.items():\n", " print('\\nKey: %s' % lora)\n", " print('Value: %s' % link)\n", " !wget -c {link} -O /content/stable-diffusion-webui/models/Lora/{lora}\n", "\n" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "pddH4PXZB-fZ" }, "outputs": [], "source": [ "#@title Run UI!\n", "%pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchtext==0.14.1 torchaudio==0.13.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu117\n", "!cd /content/stable-diffusion-webui/ && python launch.py --enable-insecure-extension-access --share" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "id": "W0k4xg3rLzO8" }, "outputs": [], "source": [ "#@title 5. Export Photos to /export\n", "%store -r rootDir \n", "\n", "from pathlib import Path\n", "import os, subprocess\n", "\n", "export_storage_dir = Path(rootDir, 'export')\n", "export_storage_dir.mkdir(exist_ok=True)\n", "\n", "!if [ $(dpkg-query -W -f='${Status}' p7zip-full 2>/dev/null | grep -c \"ok installed\") = 0 ]; then sudo apt update && sudo apt install -y p7zip-full; fi # install 7z if it isn't already installed\n", "from datetime import datetime\n", "datetime_str = datetime.now().strftime('%m-%d-%Y_%H-%M-%S')\n", "%cd \"{export_storage_dir}\"\n", "!mkdir -p \"{datetime_str}/log\"\n", "!cd \"{rootDir}/stable-diffusion-webui/log\" && mv * \"{export_storage_dir / datetime_str / 'log'}\"\n", "!cd \"{rootDir}/stable-diffusion-webui/outputs\" && mv * \"{export_storage_dir / datetime_str}\"\n", "s = subprocess.run(f'find \"{Path(export_storage_dir, datetime_str)}\" -type d -name .ipynb_checkpoints -exec rm -rv {{}} +', shell=True)\n", "!7z a -t7z -m0=lzma2 -mx=9 -mfb=64 -md=32m -ms=on \"{datetime_str}.7z\" \"{export_storage_dir / datetime_str}\"" ] } ], "metadata": { "accelerator": "GPU", "colab": { "machine_shape": "hm", "private_outputs": true, "provenance": [] }, "gpuClass": "premium", "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.9.6" }, "vscode": { "interpreter": { "hash": "31f2aee4e71d21fbe5cf8b01ff0e069b9275f58929596ceb00d14d90e3e16cd6" } } }, "nbformat": 4, "nbformat_minor": 0 } demo.launch()