stable_diffusion_1_5_webui.ipynb

{
  "cells": [
    {
      "attachments": {},
      "cell_type": "markdown",
      "metadata": {
        "id": "SKi-oMBwhutR"
      },
      "source": [
        "# AUTOMATIC1111 Stable Diffusion WebUI 1.5 + ChilloutMix Checkpoint\n",
        "\n",
        "More info: https://github.com/wibus-wee/stable_diffusion_chilloutmix_ipynb\n",
        "\n",
        "> Created by [@wibus-wee](https://github.com/wibus-wee)\n",
        ">\n",
        "> Reference: [camenduru/stable-diffusion-webui-colab](https://github.com/camenduru/stable-diffusion-webui-colab)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "from IPython.display import display\n",
        "import ipywidgets as widgets\n",
        "import requests\n",
        "\n",
        "#@title 选择模型 { display-mode: \"form\" }\n",
        "endpoint = 'https://civitai.com/api/v1/models'\n",
        "#@markdown 选择模型（ Others 需要后面填入模型ID 或 URL）\n",
        "checkpointName = 'Chilloutmix' #@param [\"Chilloutmix\", \"Sunshinemix\", \"grapefruit_hentai\", \"others\"]\n",
        "#@markdown 其他模型 （填入模型ID, 如 6424, 多个模型用英文逗号分隔）\n",
        "checkpointID = '' #@param {type:\"string\"}\n",
        "#@markdown 其他模型 （填入模型URL, 多个模型用英文逗号分隔）\n",
        "checkpointURL = '' #@param {type:\"string\"}\n",
        "\n",
        "\n",
        "defaultCheckpoint = {\n",
        "    'Chilloutmix': '6424',\n",
        "    'Sunshinemix': '9291',\n",
        "    'grapefruit_hentai': '2583'\n",
        "}\n",
        "\n",
        "downloadIds = []\n",
        "\n",
        "if checkpointID != '':\n",
        "    downloadIds = checkpointID.split(',')\n",
        "if checkpointName != 'others':\n",
        "        downloadIds.append(defaultCheckpoint[checkpointName])\n",
        "\n",
        "globalDropdowns = []\n",
        "globalVerions = []\n",
        "globalNames = []\n",
        "globalTexts = []\n",
        "checkpoints = []\n",
        "downloadLinks = []\n",
        "\n",
        "def text_on_submit(change):\n",
        "    checkpoints[checkpoints.index(change['old'])] = change['new']\n",
        "\n",
        "if checkpointURL != '':\n",
        "    _downloadLinks = checkpointURL.split(',')\n",
        "    for _downloadLink in _downloadLinks:\n",
        "        checkpoints.append(_downloadLink.split('/')[-1])\n",
        "        downloadLinks.append(_downloadLink)\n",
        "        text = widgets.Text(value=_downloadLink.split('/')[-1], description=_downloadLink.split('/')[-1], disabled=False)\n",
        "        text.observe(text_on_submit, names='value')\n",
        "        form = widgets.VBox([text])\n",
        "        display(form)\n",
        "\n",
        "def showVerionOptions(downloadId):\n",
        "    res = requests.get(endpoint + '/' + downloadId).json()\n",
        "    globalNames.append(res['name'])\n",
        "    versions = res['modelVersions']\n",
        "    globalVerions.append(versions)\n",
        "    options = []\n",
        "    for version in versions:\n",
        "        options.append(version['files'][0]['name'])\n",
        "    dropdown = widgets.Dropdown(options=options, description=res['name'])\n",
        "    globalDropdowns.append(dropdown)\n",
        "    form = widgets.VBox([dropdown])\n",
        "    display(form)\n",
        "\n",
        "for downloadId in downloadIds:\n",
        "    showVerionOptions(downloadId)\n",
        "\n",
        "def on_button_clicked(b):\n",
        "    downloadLink = None\n",
        "    for dropdown in globalDropdowns:\n",
        "        checkpoint = dropdown.value\n",
        "        versions = globalVerions[globalDropdowns.index(dropdown)]\n",
        "        for version in versions:\n",
        "            if version['files'][0]['name'] == checkpoint:\n",
        "                downloadLink = version['files'][0]['downloadUrl']\n",
        "                break\n",
        "        if downloadLink is None:\n",
        "            print('Error: downloadLink not assigned')\n",
        "            return\n",
        "        checkpoints.append(checkpoint)\n",
        "        downloadLinks.append(downloadLink)\n",
        "\n",
        "    print(\"已选择模型: \" + str(checkpoints) + \" <===> \" + str(downloadLinks))\n",
        "    %store checkpoints\n",
        "    %store downloadLinks\n",
        "\n",
        "\n",
        "button = widgets.Button(description='Use it!')\n",
        "button.on_click(on_button_clicked)\n",
        "display(button)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@title 2. 检查 GPU & 检查环境\n",
        "\n",
        "#@markdown 此步骤将检查你的 GPU 是否支持 xformers，同时会检查是否为 Paperspace 平台（仅针对 M4000 GPU 做了判断，因此付费 GPU 可能在此处将会出现逻辑判断错误的情况，你可能需要自行勾选为 Paperspace 平台）。\n",
        "\n",
        "import os, subprocess\n",
        "paperspace_m4000 = False\n",
        "#@markdown 是否为 Paperspace 平台\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: 你正在使用的是 Quadro M4000 GPU，它将无法使用 xformers。\")\n",
        "        paperspace_m4000 = True\n",
        "        isPaperspace = True\n",
        "    else:\n",
        "        print(\"你正在使用的是合适的 GPU - \" + subprocess.run(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'], stdout=subprocess.PIPE).stdout.decode('utf-8') + \"。\")\n",
        "        print(\"平台: Paperspace\" if isPaperspace else \"使用平台: Colab\")\n",
        "except:\n",
        "    print(\"似乎没有 GPU 可用。请检查你的运行时类型。\")\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. 第一次使用 - 安装依赖并启动\n",
        "\n",
        "#@markdown ## **Extensions**\n",
        "#@markdown > 你可以在安装时选择安装扩展，扩展将会在启动时自动安装。如果你想要在启动后再安装扩展，你可能需要重启运行时。\n",
        "\n",
        "#@markdown 安装 xformer 包 (仅在 Paperspace M4000 GPU 平台上运行时需要选择，若选择则会构建 xformers 包).\n",
        "xformersInstall = True #@param {type:\"boolean\"}\n",
        "#@markdown 安装 ControlNet 扩展\n",
        "controlNetExtension = True #@param {type:\"boolean\"}\n",
        "#@markdown 安装 OpenPose Editor 扩展\n",
        "openPoseExtension = True #@param {type:\"boolean\"}\n",
        "#@markdown 安装 Civitai Browser 扩展\n",
        "civitaiBrowserExtension = True #@param {type:\"boolean\"}\n",
        "#@markdown 安装 HuggingFace 扩展\n",
        "huggingFaceExtension = True #@param {type:\"boolean\"}\n",
        "#@markdown 安装 Images Browser 扩展\n",
        "imagesBrowserExtension = True #@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",
        "#@markdown ---\n",
        "\n",
        "#@markdown ## **LoRAs**\n",
        "#@markdown > 你可以在安装时选择安装模型，模型将会在启动前自动安装。\n",
        "\n",
        "#@markdown Korean Doll Likeness `v1.0`\n",
        "koreanDollLikenessv10 = True #@param {type:\"boolean\"}\n",
        "#@markdown Korean Doll Likeness `v1.5`\n",
        "koreanDollLikenessv15 = True #@param {type:\"boolean\"}\n",
        "#@markdown St. Louis (Luxurious Wheels) (Azur Lane)\n",
        "stLouis = True #@param {type:\"boolean\"}\n",
        "#@markdown Girls' Frontline-OTs-14\"lightning\"\n",
        "girlsFrontlineOTs14Lightning = True #@param {type:\"boolean\"}\n",
        "#@markdown Yae Miko | Realistic Genshin (Mixed)\n",
        "yaeMikoRealisticGenshinMixed = True #@param {type:\"boolean\"}\n",
        "#@markdown Fashion Girl\n",
        "fashionGirl = True #@param {type:\"boolean\"}\n",
        "#@markdown Cute Girl mix4\n",
        "cuteGirlMix4 = True #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown ---\n",
        "\n",
        "#@markdown ## **Textual Inversion**\n",
        "#@markdown > 你可以在安装时选择安装文本反转模型，模型将会在启动前自动安装。\n",
        "\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 ## **Others**\n",
        "\n",
        "#@markdown Styles CSV File URL\n",
        "stylesCSVFileURL = \"\" #@param {type:\"string\"}\n",
        "\n",
        "loraDownloadIDs = {\n",
        "  'koreanDollLikenessv10': 'https://github.com/wibus-wee/stable_diffusion_chilloutmix_ipynb/releases/download/koreanDollLikeness/koreanDollLikeness_v10.safetensors',\n",
        "  'koreanDollLikenessv15': 'https://github.com/wibus-wee/stable_diffusion_chilloutmix_ipynb/releases/download/koreanDollLikeness/koreanDollLikeness_v15.safetensors',\n",
        "  'stLouis': 6669,\n",
        "  'girlsFrontlineOTs14Lightning': 6525,\n",
        "  'yaeMikoRealisticGenshinMixed': 8484,\n",
        "  # 'chilloutMixss': 10850,\n",
        "  'fashionGirl': {\n",
        "    \"id\": 8217,\n",
        "    \"version\": \"v4.5\"\n",
        "  },\n",
        "  'cuteGirlMix4': 14171,\n",
        "}\n",
        "\n",
        "#@markdown ---\n",
        "\n",
        "#@markdown ## **Startup Options**\n",
        "\n",
        "#@markdown API Support\n",
        "apiSupport = True #@param {type:\"boolean\"}\n",
        "#@markdown Cors Allow Origins\n",
        "corsAllowOrigins = \"*\" #@param {type:\"string\"}\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 getLoraDownloadURLs():\n",
        "  downloadURLs = []\n",
        "  for key in loraDownloadIDs:\n",
        "    if not eval(key): # skip if not selected\n",
        "      continue\n",
        "    if type(loraDownloadIDs[key]) is int:\n",
        "      downloadURLs.append(getLatestModelDownloadURL(loraDownloadIDs[key]))\n",
        "    elif type(loraDownloadIDs[key]) is dict: # {'id': 123, 'version': 'v1.0'}\n",
        "      downloadURLs.append(getSpecificModelDownloadURL(loraDownloadIDs[key]['id'], loraDownloadIDs[key]['version']))\n",
        "    elif type(loraDownloadIDs[key]) is str: # url\n",
        "      downloadURLs.append({ 'url': loraDownloadIDs[key], 'name': loraDownloadIDs[key].split('/')[-1] })\n",
        "  downloadURLs = [x for x in downloadURLs if x is not None]\n",
        "  return downloadURLs\n",
        "\n",
        "def getTextualInversionDownloadURLs():\n",
        "  downloadURLs = []\n",
        "  for key in loraDownloadIDs:\n",
        "    if not eval(key): # skip if not selected\n",
        "      continue\n",
        "    if type(loraDownloadIDs[key]) is int:\n",
        "      downloadURLs.append(getLatestModelDownloadURL(loraDownloadIDs[key]))\n",
        "    elif type(loraDownloadIDs[key]) is dict: # {'id': 123, 'version': 'v1.0'}\n",
        "      downloadURLs.append(getSpecificModelDownloadURL(loraDownloadIDs[key]['id'], loraDownloadIDs[key]['version']))\n",
        "    elif type(loraDownloadIDs[key]) is str: # url\n",
        "      downloadURLs.append({ 'url': loraDownloadIDs[key], 'name': loraDownloadIDs[key].split('/')[-1] })\n",
        "  downloadURLs = [x for x in downloadURLs if x is not None]\n",
        "  return downloadURLs\n",
        "    \n",
        "\n",
        "loraDownloadURLs = getLoraDownloadURLs()\n",
        "textualInversionDownloadURLs = getTextualInversionDownloadURLs()\n",
        "\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/camenduru/stable-diffusion-webui {rootDir}/stable-diffusion-webui\n",
        "!wget https://raw.githubusercontent.com/camenduru/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",
        "autoDetectDownload(downloadLinks, checkpoints, rootDir + \"/stable-diffusion-webui/models/Stable-diffusion\")\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 loraDownloadURLs:\n",
        "  autoDetectDownload(model[\"url\"], model[\"name\"], rootDir + \"/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora\")\n",
        "for model in textualInversionDownloadURLs:\n",
        "  autoDetectDownload(model[\"url\"], model[\"name\"], rootDir + \"/stable-diffusion-webui/embeddings\")\n",
        "if stylesCSVFileURL:\n",
        "  autoDetectDownload(stylesCSVFileURL, \"styles.csv\", rootDir + \"/stable-diffusion-webui\")\n",
        "\n",
        "if additionalNetworksExtension:\n",
        "  !rm -rf {rootDir}/stable-diffusion-webui/models/Lora\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",
        "!python launch.py --share --enable-insecure-extension-access {'' if paperspace_m4000 and not xformersInstall else '--xformers'} {'--api --cors-allow-origins \"'corsAllowOrigins'\" --listen' if apiSupport else ''}"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "UilfY37zpOoK"
      },
      "outputs": [],
      "source": [
        "\n",
        "#@title 4. 重启 - 重新启动 Stable Diffusion WebUI\n",
        "\n",
        "%store -r paperspace_m4000 \n",
        "%store -r rootDir \n",
        "%store -r stableDiffusionWebUIInstalled\n",
        "\n",
        "if not stableDiffusionWebUIInstalled:\n",
        "    print(\"请先运行安装 Stable Diffusion WebUI 的单元格。\")\n",
        "    exit()\n",
        "\n",
        "if additionalNetworksExtension:\n",
        "  !rm -rf {rootDir}/stable-diffusion-webui/models/Lora\n",
        "  !ln -s {rootDir}/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora {rootDir}/stable-diffusion-webui/models/Lora\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",
        "!python launch.py --share --enable-insecure-extension-access {'' if paperspace_m4000 and not xformersInstall else '--xformers'} {'--api --cors-allow-origins \"'corsAllowOrigins'\" --listen' if apiSupport else ''}"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@title 5. 导出 - 导出生成的图片\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": {
      "collapsed_sections": [
        "SEQh1nJopSDr"
      ],
      "private_outputs": true,
      "provenance": []
    },
    "gpuClass": "standard",
    "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.11.1"
    },
    "vscode": {
      "interpreter": {
        "hash": "5c7b89af1651d0b8571dde13640ecdccf7d5a6204171d6ab33e7c296e100e08a"
      }
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}