230627.ipynb

{
  "cells": [
    {
      "attachments": {},
      "cell_type": "markdown",
      "metadata": {
        "colab_type": "text",
        "id": "view-in-github"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/ennnnny/sd_colab/blob/self/230627.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "attachments": {},
      "cell_type": "markdown",
      "metadata": {
        "id": "Yl3WhkeHZDRq"
      },
      "source": [
        "[![GitHub Repo stars](https://img.shields.io/github/stars/ennnnny/sd_colab?style=social)](https://github.com/ennnnny/sd_colab)"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "2EmivlLQZLRz"
      },
      "source": [
        "# **点击播放按钮一键运行再点击左侧图标展开**\n",
        "\n",
        "---\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "6hHNXtOEiKJy"
      },
      "source": [
        "##**展开第四步出现网址即可跳转**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "00QkTUUZPrnb"
      },
      "outputs": [],
      "source": [
        "#@title ## 1、加载谷歌云盘并初始化环境\n",
        "#@markdown ####是否允许加载谷歌云盘（禁止将无法保存模型）：##\n",
        "is_load_pan = \"\\u5141\\u8BB8\" #@param [\"允许\",\"禁止\"]\n",
        "import os\n",
        "import concurrent.futures\n",
        "def init_pan():\n",
        "  if is_load_pan==\"允许\":\n",
        "    from google.colab import drive\n",
        "    drive.mount('/content/drive')\n",
        "  else:\n",
        "    print('已禁止加载云盘')\n",
        "def aira2_install():\n",
        "  !apt-get -y install -qq aria2\n",
        "executor = concurrent.futures.ThreadPoolExecutor(max_workers=2)\n",
        "task1 = executor.submit(init_pan)\n",
        "task2 = executor.submit(aira2_install)\n",
        "concurrent.futures.wait([task1,task2])\n",
        "\n",
        "#@markdown ####大模型路径：##\n",
        "ckpt_path = \"/content/drive/MyDrive/ckpt\" #@param {type: \"string\"}\n",
        "if is_load_pan==\"禁止\":\n",
        "  ckpt_path = \"\"\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "SaAJk33ppFw1"
      },
      "outputs": [],
      "source": [
        "#@title ## 2.0、下载各种插件\n",
        "str_pre1=\"S\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"\n",
        "str_pre2=\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"\n",
        "sd0=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+str_pre2+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"\n",
        "sd=sd0+'i'\n",
        "wi='w'+'e'+'b'+'u'+'i'\n",
        "wi2='e'+'b'+'u'+'i'\n",
        "str_temp1=str_pre1+'-d'+str_pre2\n",
        "str_temp2='sd'+'-'+wi\n",
        "str_temp3=str_pre1+'-D'+str_pre2+'-W'+wi2\n",
        "str_temp4=str_pre1+'D'+str_pre2\n",
        "#@markdown ####初始化的大模型：##\n",
        "first_checkpoint = \"none\" #@param [\"Dark_sushi_mix.safetensors\", \"AnythingV5V3_v5PrtRE.safetensors\", \"chilloutmix_NiPrunedFp16Fix.safetensors\", \"rpg_V4.safetensors\", \"protogenV22Anime_22.safetensors\",\"none\"]\n",
        "#@markdown ####stable diffusion的ui页面：##\n",
        "ssdd_ui = \"AUTOMATIC1111\\u539F\\u7248\\u754C\\u9762\" #@param [\"AUTOMATIC1111原版界面\",\"anapnoe手机端完美适配\"]\n",
        "sd_web_kitchen_theme = True  # @param {type:'boolean'}\n",
        "sd_mov = False # @param {type:'boolean'}\n",
        "sd_segment_anything = False # @param {type:'boolean'}\n",
        "sd_adetailer = True # @param {type:'boolean'}\n",
        "sd_llul = False # @param {type:'boolean'}\n",
        "sd_regional_prompter = False # @param {type:'boolean'}\n",
        "sd_block_weight = False # @param {type:'boolean'}\n",
        "sd_dynamic_thresholding = False # @param {type:'boolean'}\n",
        "sd_deoldify = False # @param {type:'boolean'}\n",
        "sd_roop = False # @param {type:'boolean'}\n",
        "sd_faceswaplab = False # @param {type:'boolean'}\n",
        "sd_cutoff = False # @param {type:'boolean'}\n",
        "sd_qrcode = False # @param {type:'boolean'}\n",
        "sd_qrcode_check = False # @param {type:'boolean'}\n",
        "sd_openpose_editor = False # @param {type:'boolean'}\n",
        "sd_photopea = False # @param {type:'boolean'}\n",
        "sd_agent_scheduler = False # @param {type:'boolean'}\n",
        "sd_comfyui = False # @param {type:'boolean'}\n",
        "\n",
        "sd_tunnels = False # @param {type:'boolean'}\n",
        "sd_SadTalker = False # @param {type:'boolean'}\n",
        "\n",
        "\n",
        "if 'ckpt_path' in globals() and ckpt_path!=\"\":\n",
        "  n_ckpt_path = f\"{ckpt_path}\"\n",
        "else:\n",
        "  n_ckpt_path = f\"/content/{sd}/models/{str_temp1}\"\n",
        "\n",
        "#@markdown ####是否加载云盘里的extensions、VAE、embeddings、lora、checkpoint？##\n",
        "is_pan_extensions = True  # @param {type:'boolean'}\n",
        "%cd /content/\n",
        "if ssdd_ui==\"anapnoe手机端完美适配\":\n",
        "  sd_web_kitchen_theme = False\n",
        "  !git clone https://github.com/anapnoe/{sd}\n",
        "else:\n",
        "  !git clone -b v1.6.0 --single-branch https://github.com/AUTOMATIC1111/{sd}\n",
        "import concurrent.futures\n",
        "import time\n",
        "import subprocess\n",
        "\n",
        "checkpoint_url = {\n",
        "    \"Dark_sushi_mix.safetensors\"     : \"https://huggingface.co/mdl-mirror/dark-sushi-mix/resolve/main/darkSushiMixMix_brighter.safetensors\",\n",
        "    \"AnythingV5V3_v5PrtRE.safetensors\"     : \"https://huggingface.co/ckpt/anything-v5.0/resolve/main/AnythingV5V3_v5PrtRE.safetensors\",\n",
        "    \"chilloutmix_NiPrunedFp16Fix.safetensors\"      : \"https://huggingface.co/naonovn/chilloutmix_NiPrunedFp32Fix/resolve/main/chilloutmix_NiPrunedFp32Fix.safetensors\",\n",
        "    \"rpg_V4.safetensors\"         : \"https://huggingface.co/Anashel/rpg/resolve/main/RPG-V4-Model-Download/RPG-v4.safetensors\",\n",
        "    \"ProtoGen_X5.8-pruned-fp16.safetensors\"         : \"https://huggingface.co/darkstorm2150/Protogen_x5.8_Official_Release/resolve/main/ProtoGen_X5.8-pruned-fp16.safetensors\",\n",
        "    \"none\"         : \"\",\n",
        "}\n",
        "\n",
        "def run_git_download():\n",
        "  start_time = time.time()\n",
        "  !git clone https://github.com/Physton/sd-{wi}-prompt-all-in-one /content/{sd}/extensions/sd-{wi}-prompt-all-in-one\n",
        "  !git clone https://github.com/Mikubill/sd-{wi}-controlnet /content/{sd}/extensions/sd-{wi}-controlnet\n",
        "  !git clone https://github.com/dtlnor/{sd}-localization-zh_CN /content/{sd}/extensions/{sd}-localization-zh_CN\n",
        "  !git clone https://github.com/hanamizuki-ai/{sd}-localization-zh_Hans /content/{sd}/extensions/{sd}-localization-zh_Hans\n",
        "  !git clone https://github.com/fkunn1326/openpose-editor /content/{sd}/extensions/openpose-editor\n",
        "  !git clone https://github.com/DominikDoom/a1111-sd-{wi}-tagcomplete /content/{sd}/extensions/a1111-sd-{wi}-tagcomplete\n",
        "  !git clone https://github.com/Coyote-A/ultimate-upscale-for-automatic1111 /content/{sd}/extensions/ultimate-upscale\n",
        "  !git clone https://github.com/toriato/{sd}-wd14-tagger /content/{sd}/extensions/{sd}-wd14-tagger\n",
        "  !git clone https://github.com/nonnonstop/{str_temp2}-3d-open-pose-editor /content/{sd}/extensions/{str_temp2}-3d-open-pose-editor\n",
        "  !git clone https://github.com/hako-mikan/{str_temp2}-lora-block-weight /content/{sd}/extensions/{str_temp2}-lora-block-weight\n",
        "  !git clone https://github.com/kohya-ss/{str_temp2}-additional-networks /content/{sd}/extensions/{str_temp2}-additional-networks\n",
        "  !git clone https://github.com/AbdullahAlfaraj/Auto-Photoshop-{str_temp4}-Plugin /content/{sd}/extensions/Auto-Photoshop-{str_temp4}-Plugin\n",
        "  !git clone https://github.com/AUTOMATIC1111/{sd}-rembg /content/{sd}/extensions/{sd}-rembg\n",
        "  !git clone https://github.com/zanllp/{str_temp2}-infinite-image-browsing /content/{sd}/extensions/{str_temp2}-infinite-image-browsing\n",
        "  !git clone https://github.com/butaixianran/{str_temp3}-Civitai-Helper /content/{sd}/extensions/{str_temp3}-Civitai-Helper\n",
        "\n",
        "  if sd_web_kitchen_theme:\n",
        "    !git clone https://github.com/canisminor1990/{str_temp2}-kitchen-theme /content/{sd}/extensions/kitchen-theme\n",
        "  if sd_mov:\n",
        "    !git clone https://github.com/Scholar01/{str_temp2}-mov2mov /content/{sd}/extensions/{str_temp2}-mov2mov\n",
        "  if sd_segment_anything:\n",
        "    !git clone https://github.com/continue-revolution/{str_temp2}-segment-anything /content/{sd}/extensions/a1111-{str_temp2}-segment-anything\n",
        "  if sd_adetailer:\n",
        "    !git clone https://github.com/Bing-su/adetailer /content/{sd}/extensions/adetailer\n",
        "  if sd_llul:\n",
        "    !git clone https://github.com/hnmr293/{str_temp2}-llul.git /content/{sd}/extensions/{str_temp2}-llul\n",
        "  if sd_tunnels:\n",
        "    !git clone https://github.com/camenduru/{str_temp2}-tunnels /content/{sd}/extensions/{str_temp2}-tunnels\n",
        "  if sd_SadTalker:\n",
        "    !git clone https://github.com/Winfredy/SadTalker.git /content/{sd}/extensions/SadTalker\n",
        "  if sd_regional_prompter:\n",
        "    !git clone https://github.com/hako-mikan/{str_temp2}-regional-prompter /content/{sd}/extensions/{str_temp2}-regional-prompter\n",
        "  if sd_block_weight:\n",
        "    !git clone https://github.com/hako-mikan/{str_temp2}-lora-block-weight /content/{sd}/extensions/{str_temp2}-lora-block-weight\n",
        "  if sd_dynamic_thresholding:\n",
        "    !git clone https://github.com/mcmonkeyprojects/sd-dynamic-thresholding /content/{sd}/extensions/sd-dynamic-thresholding\n",
        "  if sd_roop:\n",
        "    !git clone https://github.com/s0md3v/{str_temp2}-roop /content/{sd}/extensions/{str_temp2}-roop\n",
        "  if sd_cutoff:\n",
        "    !git clone https://github.com/hnmr293/{str_temp2}-cutoff /content/{sd}/extensions/{str_temp2}-cutoff\n",
        "  if sd_qrcode:\n",
        "    !git clone https://github.com/antfu/{str_temp2}-qrcode-toolkit /content/{sd}/extensions/{str_temp2}-qrcode-toolkit\n",
        "  if sd_openpose_editor:\n",
        "    !git clone https://github.com/huchenlei/{str_temp2}-openpose-editor /content/{sd}/extensions/{str_temp2}-openpose-editor\n",
        "  if sd_photopea:\n",
        "    !git clone https://github.com/yankooliveira/{str_temp2}-photopea-embed /content/{sd}/extensions/{str_temp2}-photopea-embed\n",
        "  if sd_agent_scheduler:\n",
        "    !git clone https://github.com/ArtVentureX/{str_temp2}-agent-scheduler /content/{sd}/extensions/{str_temp2}-agent-scheduler\n",
        "  if sd_deoldify:\n",
        "    !git clone https://github.com/SpenserCai/{str_temp2}-deoldify /content/{sd}/extensions/{str_temp2}-deoldify\n",
        "  if sd_comfyui:\n",
        "    !git clone https://github.com/ModelSurge/{str_temp2}-comfyui /content/{sd}/extensions/{str_temp2}-comfyui\n",
        "  if sd_faceswaplab:\n",
        "    !git clone https://github.com/glucauze/{str_temp2}-faceswaplab /content/{sd}/extensions/{str_temp2}-faceswaplab\n",
        "  if sd_qrcode_check:\n",
        "    !git clone https://github.com/design61/{str_temp2}-qrcode-checkbtn /content/{sd}/extensions/{str_temp2}-qrcode-checkbtn\n",
        "  \n",
        "  #xformers加速\n",
        "  end_time = time.time()\n",
        "  print(\"已克隆git耗时：\", end_time-start_time, \"秒\")\n",
        "\n",
        "def run_aria2c_download():\n",
        "  start_time = time.time()\n",
        "  cmd=f\"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {checkpoint_url[first_checkpoint]}  -d {n_ckpt_path} -o {first_checkpoint}\"\n",
        "  subprocess.run(cmd, shell=True)\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11e_sd15_ip2p.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11e_sd15_shuffle.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11f1e_sd15_tile.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11f1p_sd15_depth.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_canny_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_canny.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_inpaint.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_lineart_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_lineart.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_mlsd.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_normalbae.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_openpose_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_openpose.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_scribble_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_scribble.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_seg_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_seg.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_softedge_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15_softedge.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11p_sd15s2_lineart_anime.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11u_sd15_tile_fp16.safetensors -d /content/{sd}/models/ControlNet -o control_v11u_sd15_tile.safetensors\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lokCX/4x-Ultrasharp/resolve/main/4x-UltraSharp.pth -d /content/{sd}/models/ESRGAN/ -o 4x-UltraSharp.pth\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/datasets/daasd/CN.csv/resolve/main/CN.csv -d /content/{sd}/extensions/a1111-{str_temp2}-tagcomplete/tags -o CN.csv\n",
        "  if sd_qrcode:\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/DionTimmer/controlnet_qrcode/resolve/main/control_v1p_sd15_qrcode.safetensors -d /content/{sd}/models/ControlNet -o control_v1p_sd15_qrcode.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ioclab/ioc-controlnet/resolve/main/models/control_v1p_sd15_brightness.safetensors -d /content/{sd}/models/ControlNet -o control_v1p_sd15_brightness.safetensors\n",
        "  if sd_segment_anything:\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -d /content/{sd}/models/sam -o sam_vit_h_4b8939.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ShilongLiu/GroundingDINO/resolve/main/groundingdino_swinb_cogcoor.pth -d /content/{sd}/extensions/a1111-{str_temp2}-segment-anything/models/grounding-dino -o groundingdino_swinb_cogcoor.pth\n",
        "  if sd_adetailer:\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt -d /content/{sd}/models/adetailer -o deepfashion2_yolov8s-seg.pt\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8m.pt -d /content/{sd}/models/adetailer -o face_yolov8m.pt\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/Bingsu/adetailer/resolve/main/hand_yolov8s.pt -d /content/{sd}/models/adetailer -o hand_yolov8s.pt\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/Bingsu/adetailer/resolve/main/person_yolov8m-seg.pt -d /content/{sd}/models/adetailer -o person_yolov8m-seg.pt\n",
        "  if sd_SadTalker:\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00109-model.pth.tar -d /content/{sd}/extensions/SadTalker/checkpoints -o mapping_00109-model.pth.tar\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00229-model.pth.tar -d /content/{sd}/extensions/SadTalker/checkpoints -o mapping_00229-model.pth.tar\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_256.safetensors -d /content/{sd}/extensions/SadTalker/checkpoints -o SadTalker_V0.0.2_256.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_512.safetensors -d /content/{sd}/extensions/SadTalker/checkpoints -o SadTalker_V0.0.2_512.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/xinntao/facexlib/releases/download/v0.1.0/alignment_WFLW_4HG.pth -d /content/{sd}/extensions/SadTalker/gfpgan/weights -o alignment_WFLW_4HG.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth -d /content/{sd}/extensions/SadTalker/gfpgan/weights -o detection_Resnet50_Final.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -d /content/{sd}/extensions/SadTalker/gfpgan/weights -o GFPGANv1.4.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth -d /content/{sd}/extensions/SadTalker/gfpgan/weights -o parsing_parsenet.pth\n",
        "  if sd_deoldify:\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/spensercai/DeOldify/resolve/main/ColorizeArtistic_gen.pth -d /content/{sd}/models/deoldify -o ColorizeArtistic_gen.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/spensercai/DeOldify/resolve/main/ColorizeStable_gen.pth -d /content/{sd}/models/deoldify -o ColorizeStable_gen.pth\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/spensercai/DeOldify/resolve/main/ColorizeVideo_gen.pth -d /content/{sd}/models/deoldify -o ColorizeVideo_gen.pth\n",
        "\n",
        "  end_time = time.time()\n",
        "  print(\"aria2c完成下载耗时：\", end_time-start_time, \"秒\")\n",
        "\n",
        "def curl_download():\n",
        "  start_time = time.time()\n",
        "  !curl -Lo \"/content/{sd}/models/VAE/vae-ft-mse-840000-ema-pruned.safetensors\" https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors\n",
        "  !curl -Lo \"/content/{sd}/models/VAE/kl-f8-anime2.ckpt\" https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/4c4f05104055c029ad577c18ac176462f0d1d7c1/vae/kl-f8-anime2.ckpt\n",
        "  !curl -Lo \"/content/{sd}/models/VAE/animevae.pt\" https://huggingface.co/swl-models/animvae/resolve/main/animevae.pt\n",
        "  !curl -Lo \"/content/{sd}/models/VAE/sdxl_vae.safetensors\" https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors\n",
        "  end_time = time.time()\n",
        "  print(\"curl完成下载耗时：\", end_time-start_time,\"秒\")\n",
        "\n",
        "def wget_download():\n",
        "  start_time = time.time()\n",
        "  !apt install libunwind8-dev -yqq\n",
        "  os.environ[\"LD_PRELOAD\"] = \"libtcmalloc.so.4\"\n",
        "  os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
        "  !sudo apt-get install sox ffmpeg libcairo2 libcairo2-dev\n",
        "  end_time = time.time()\n",
        "  print(\"wget完成下载耗时：\", end_time-start_time,\"秒\")\n",
        "\n",
        "def pip_download():\n",
        "  start_time = time.time()\n",
        "  !pip install xformers xformers==0.0.20\n",
        "  if sd_SadTalker:\n",
        "    !pip install -r /content/{sd}/extensions/SadTalker/requirements.txt\n",
        "  if sd_faceswaplab:\n",
        "    !pip install protobuf==3.20.2\n",
        "\n",
        "  end_time = time.time()\n",
        "  print(\"pip完成下载耗时：\", end_time-start_time,\"秒\")\n",
        "\n",
        "executor = concurrent.futures.ThreadPoolExecutor(max_workers=5)\n",
        "task1 = executor.submit(run_git_download)\n",
        "task2 = executor.submit(run_aria2c_download)\n",
        "task3 = executor.submit(curl_download)\n",
        "task4 = executor.submit(wget_download)\n",
        "task5 = executor.submit(pip_download)\n",
        "concurrent.futures.wait([task1])\n",
        "concurrent.futures.wait([task2,task3,task4,task5])\n",
        "\n",
        "%cd /content/{sd}\n",
        "import shutil\n",
        "if os.path.exists(f'/content/{sd}/embeddings'):\n",
        "  shutil.rmtree(f'/content/{sd}/embeddings')\n",
        "!git clone https://huggingface.co/nolanaatama/embeddings\n",
        "if os.path.exists(f'/content/{sd}/embeddings/.git'):\n",
        "  shutil.rmtree(f'/content/{sd}/embeddings/.git')\n",
        "\n",
        "#个人插件从云盘的extensions文件夹与VAE文件夹加载\n",
        "if is_pan_extensions:\n",
        "  if os.path.exists(\"/content/drive/MyDrive/extensions\"):\n",
        "    !rsync -a /content/drive/MyDrive/extensions/* /content/{sd}/extensions\n",
        "    print('已加载云盘里的插件')\n",
        "  if os.path.exists(\"/content/drive/MyDrive/VAE\"):\n",
        "    !rsync -a /content/drive/MyDrive/VAE/* /content/{sd}/models/VAE\n",
        "    print('已加载云盘里的VAE')\n",
        "  if os.path.exists(\"/content/drive/MyDrive/embeddings\"):\n",
        "    !rsync -a /content/drive/MyDrive/embeddings/* /content/{sd}/embeddings\n",
        "    print('已加载云盘里的embeddings')\n",
        "  if os.path.exists(\"/content/drive/MyDrive/lora\"):\n",
        "    !mkdir -p /content/{sd}/models/Lora\n",
        "    !rsync -a /content/drive/MyDrive/lora/* /content/{sd}/models/Lora\n",
        "    print('已加载云盘里的lora')\n",
        "  if os.path.exists(\"/content/drive/MyDrive/checkpoint\"):\n",
        "    !rsync -a /content/drive/MyDrive/checkpoint/* /content/{sd}/models/Stable-diffusion\n",
        "    print('已加载云盘里的Stable-diffusion')\n",
        "\n",
        "# if sd_SadTalker:\n",
        "#   print('下载SadTalker模型')\n",
        "#   !wget -nc https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00109-model.pth.tar -O  /content/{sd}/extensions/SadTalker/checkpoints/mapping_00109-model.pth.tar\n",
        "#   !wget -nc https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00229-model.pth.tar -O  /content/{sd}/extensions/SadTalker/checkpoints/mapping_00229-model.pth.tar\n",
        "#   !wget -nc https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_256.safetensors -O  /content/{sd}/extensions/SadTalker/checkpoints/SadTalker_V0.0.2_256.safetensors\n",
        "#   !wget -nc https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_512.safetensors -O  /content/{sd}/extensions/SadTalker/checkpoints/SadTalker_V0.0.2_512.safetensors\n",
        "#   !wget -nc https://github.com/xinntao/facexlib/releases/download/v0.1.0/alignment_WFLW_4HG.pth -O /content/{sd}/extensions/SadTalker/gfpgan/weights/alignment_WFLW_4HG.pth \n",
        "#   !wget -nc https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth -O /content/{sd}/extensions/SadTalker/gfpgan/weights/detection_Resnet50_Final.pth \n",
        "#   !wget -nc https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -O /content/{sd}/extensions/SadTalker/gfpgan/weights/GFPGANv1.4.pth \n",
        "#   !wget -nc https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth -O /content/{sd}/extensions/SadTalker/gfpgan/weights/parsing_parsenet.pth\n",
        "#   !pip install -r /content/{sd}/extensions/SadTalker/requirements.txt\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "seJAeop9oO3b"
      },
      "outputs": [],
      "source": [
        "#@title ## 2.1、下载列表管理/增删改查模型或其他文件\n",
        "#@markdown ####选择mod集合(如果云盘没找到集合则在根目录创建)：##\n",
        "mod_json_name = \"model.json\" #@param {type: \"string\"}\n",
        "# 创建并写入文本到文件\n",
        "mod_json_dir=''\n",
        "\n",
        "#递归目录查找mod集合\n",
        "def find_model_json(root_path,mod_json_name):\n",
        "    for file_name in os.listdir(root_path):\n",
        "        file_path = os.path.join(root_path, file_name)\n",
        "        if os.path.isdir(file_path):\n",
        "            res=find_model_json(file_path,mod_json_name)\n",
        "            if res is not None:\n",
        "              return res\n",
        "        elif file_name == mod_json_name:\n",
        "            print(\"找到model.json文件：\", file_path)\n",
        "            return file_path\n",
        "\n",
        "if 'is_load_pan' in globals() and is_load_pan==\"允许\":\n",
        "  mod_json_dir=find_model_json('/content/drive/',mod_json_name)\n",
        "  if mod_json_dir==None:\n",
        "    with open(f'/content/drive/MyDrive/{mod_json_name}', 'w') as f:\n",
        "      f.write('[]')\n",
        "      mod_json_dir=f'/content/drive/MyDrive/{mod_json_name}'\n",
        "else:\n",
        "  mod_json_dir=f'/content/{mod_json_name}'\n",
        "\n",
        "import json\n",
        "import pandas as pd\n",
        "# drive.mount('/content/drive')\n",
        "from IPython.display import display, clear_output, HTML\n",
        "import ipywidgets as widgets\n",
        "from datetime import datetime\n",
        "import time\n",
        "\n",
        "custom_css = \"\"\"\n",
        "<style>\n",
        "    .custom-button {\n",
        "        padding: 10px 25px;\n",
        "    }\n",
        "    .custom-button.save {\n",
        "        background-color: #67C23A;\n",
        "        float:right\n",
        "    }\n",
        "    .custom-button.add {\n",
        "        background-color: #409EFF;\n",
        "    }\n",
        "    .custom-button.delete {\n",
        "        background-color: #F56C6C;\n",
        "    }\n",
        "    .custom-button.update {\n",
        "        background-color: #E6A23C;\n",
        "    }\n",
        "</style>\n",
        "\"\"\"\n",
        "display(HTML(custom_css))\n",
        "\n",
        "# 加载JSON数据为Python对象\n",
        "try:\n",
        "    # 尝试打开你的文件\n",
        "    with open(mod_json_dir, 'r') as f:\n",
        "        try:\n",
        "            # 尝试使用json模块的load()函数，将文件内容转换为JSON\n",
        "            data = json.load(f)\n",
        "        except json.JSONDecodeError:\n",
        "            # 如果出现 JSONDecodeError 异常，设置 data 为一个空字典\n",
        "            data = {}\n",
        "except FileNotFoundError:\n",
        "    # 如果出现 FileNotFoundError 异常，设置 data 为一个空字典\n",
        "    data = {}\n",
        "\n",
        "print(mod_json_dir)\n",
        "print(data)\n",
        "# 将JSON数据转换为Pandas DataFrame\n",
        "df = pd.DataFrame(data)\n",
        "\n",
        "# 显示表格\n",
        "def show_table():\n",
        "    clear_output()\n",
        "    # 新字段兼容\n",
        "    if not df.columns.empty and 'path' not in df.columns:\n",
        "      df['path'] = ''\n",
        "    styled_df = df.style.set_properties(**{'text-align': 'center'})\n",
        "\n",
        "    # 设置列宽度和居中对齐\n",
        "    styled_df.set_table_styles([{'selector': 'th', 'props': [('max-width', '150px'), ('text-align', 'center')]}])\n",
        "\n",
        "    # 输出DataFrame\n",
        "    display(styled_df)\n",
        "    display(widgets.HBox([input_name, input_type, input_download_link, input_size, input_path]))\n",
        "    display(widgets.HBox([input_index, read_button, update_button, delete_button, add_button]))\n",
        "    display(widgets.HBox([save_button]))\n",
        "\n",
        "# 添加一行数据\n",
        "def add_row(button):\n",
        "  try:\n",
        "    if not input_download_link.value:\n",
        "      display(HTML(f\"<p style='color: red;'>下载链接为必填项</p>\"))\n",
        "      return\n",
        "\n",
        "    new_row = {\n",
        "        \"name\": input_name.value.strip() or '未命名',\n",
        "        \"type\": '未定义' if input_type.value == '不选则根据下载内容决定' else input_type.value,\n",
        "        \"downloadLink\": input_download_link.value.strip(), # 修改这里的键名\n",
        "        \"size\": input_size.value,\n",
        "        \"path\": input_path.value,\n",
        "        \"lastChangeDate\": datetime.now().strftime('%Y-%m-%d')\n",
        "    }\n",
        "    global df\n",
        "    df = df.append(new_row, ignore_index=True)\n",
        "    lock_buttons()\n",
        "    show_table()\n",
        "    clear_input()\n",
        "    display(HTML(\"<p id='add-msg' style='color: green;'>新增{}成功</p>\".format(new_row)))\n",
        "    time.sleep(1)\n",
        "    display(HTML(\"<script>document.getElementById('add-msg').remove()</script>\"))\n",
        "  except Exception as e:\n",
        "    display(HTML(f\"<p style='color: red;'>新增出错：{e}</p>\"))\n",
        "\n",
        "# 删除一行数据\n",
        "def delete_row(button):\n",
        "  try:\n",
        "    index = int(input_index.value)\n",
        "    global df\n",
        "    df = df.drop(index)\n",
        "    lock_buttons()\n",
        "    show_table()\n",
        "    display(HTML(\"<p id='delete-msg' style='color: green;'>删除{}成功</p>\".format(index)))\n",
        "    time.sleep(1)\n",
        "    display(HTML(\"<script>document.getElementById('delete-msg').remove()</script>\"))\n",
        "  except Exception as e:\n",
        "    display(HTML(f\"<p style='color: red;'>删除出错：{e}</p>\"))\n",
        "\n",
        "#读取一行数据\n",
        "def read_row(button):\n",
        "    try:\n",
        "        index = int(input_index.value)\n",
        "        input_name.value = df.at[index, 'name']\n",
        "        input_type.value ='不选则根据下载内容决定' if df.at[index, 'type']=='未定义' else df.at[index, 'type']\n",
        "        input_download_link.value = df.at[index, 'downloadLink']\n",
        "        input_size.value = df.at[index, 'size']\n",
        "        input_path.value = df.at[index, 'path']\n",
        "        display(HTML(f\"<p id='read-msg' style='color: green;'>读取{index}成功</p>\"))\n",
        "        time.sleep(1)\n",
        "        display(HTML(\"<script>document.getElementById('read-msg').remove()</script>\"))\n",
        "    except Exception as e:\n",
        "        display(HTML(f\"<p style='color: red;'>读取出错：{e}</p>\"))\n",
        "# 修改一行数据\n",
        "def update_row(button):\n",
        "    try:\n",
        "      index = int(input_index.value)\n",
        "      if input_download_link.value:\n",
        "        df.at[index, 'downloadLink'] = input_download_link.value.strip()\n",
        "      else:\n",
        "        display(HTML(f\"<p style='color: red;'>下载链接为必填项{e}</p>\"))\n",
        "        return\n",
        "      df.at[index, 'name'] = input_name.value.strip() or '未命名'\n",
        "      df.at[index, 'type'] = '未定义' if input_type.value == '不选则根据下载内容决定' else input_type.value\n",
        "      df.at[index, 'size'] = input_size.value\n",
        "      df.at[index, 'path'] = input_path.value or ''\n",
        "      df.at[index, 'lastChangeDate'] = datetime.now().strftime('%Y-%m-%d')\n",
        "      lock_buttons()\n",
        "      clear_input()\n",
        "      show_table()\n",
        "      display(HTML(\"<p id='update-msg' style='color: green;'>修改{}成功</p>\".format(index)))\n",
        "      time.sleep(1)\n",
        "      display(HTML(\"<script>document.getElementById('update-msg').remove()</script>\"))\n",
        "    except Exception as e:\n",
        "      display(HTML(f\"<p style='color: red;'>修改出错：{e}</p>\"))\n",
        "\n",
        "#清空输入框\n",
        "def clear_input():\n",
        "    try:\n",
        "        input_name.value = ''\n",
        "        input_type.value = ''\n",
        "        input_download_link.value = ''\n",
        "        input_size.value = ''\n",
        "        input_path.value = ''\n",
        "    except Exception as e:\n",
        "        display(HTML(f\"<p style='color: red;'>重置输入框出错：{e}</p>\"))\n",
        "\n",
        "#保存\n",
        "def save_row(button):\n",
        "    global df\n",
        "    try:\n",
        "        with open(mod_json_dir, 'w') as Wfile:\n",
        "            json=df.to_json(orient='records')\n",
        "            Wfile.write(json)\n",
        "            display(HTML(\"<p id='save-msg' style='color: green;'>保存成功</p>\"))\n",
        "            time.sleep(1)\n",
        "            display(HTML(\"<script>document.getElementById('save-msg').remove()</script>\"))\n",
        "    except Exception as e:\n",
        "        display(HTML(f\"<p style='color: red;'>保存出错：{e}</p>\"))\n",
        "\n",
        "\n",
        "\n",
        "# 定义输入框和按钮\n",
        "input_name = widgets.Text(description='名字',placeholder='不填则根据下载内容决定')\n",
        "input_type = widgets.Text(description='类型',placeholder='选填')\n",
        "input_download_link = widgets.Text(description='下载链接',placeholder='必填：C站或者huggingface') # 修改这里的描述\n",
        "input_size = widgets.Text(description='大小',placeholder='选填')\n",
        "input_path = widgets.Text(description='下载路径',placeholder='除了lora和checkpoint其余必填')\n",
        "input_index = widgets.Text(description='mod编号',placeholder='与新增无关，用于读改删mod',layout=widgets.Layout(margin='10px'))\n",
        "add_button = widgets.Button(description='新增mod', layout=widgets.Layout(margin='10px 180px'), button_style='info')\n",
        "delete_button = widgets.Button(description='删除mod', layout=widgets.Layout(margin='10px 10px'), button_style='danger')\n",
        "update_button = widgets.Button(description='更新mod', layout=widgets.Layout(margin='10px 30px 10px 0'), button_style='warning')\n",
        "save_button = widgets.Button(description='保存', layout=widgets.Layout(margin='20px auto'), button_style='success')\n",
        "read_button = widgets.Button(description='读取mod', layout=widgets.Layout(margin='10px 0px 10px 30px'), button_style='primary')\n",
        "\n",
        "\n",
        "#节流\n",
        "def lock_buttons():\n",
        "    # add_button.disabled = True\n",
        "    delete_button.disabled = True\n",
        "    update_button.disabled = True\n",
        "\n",
        "def unlock_buttons():\n",
        "    # add_button.disabled = False\n",
        "    delete_button.disabled = False\n",
        "    update_button.disabled = False\n",
        "\n",
        "def on_input_change(change):\n",
        "  if change.new:\n",
        "    # len(change.new)\n",
        "    if change.new==change.owner.value:\n",
        "      if len(change.new)==1:\n",
        "        lock_buttons()\n",
        "        time.sleep(0.5)\n",
        "        unlock_buttons()\n",
        "  else:\n",
        "    lock_buttons()\n",
        "\n",
        "\n",
        "# 为输入框添加输入事件\n",
        "# input_name.observe(on_input_change, names='value')\n",
        "# input_type.observe(on_input_change, names='value')\n",
        "# input_download_link.observe(on_input_change, names='value')\n",
        "# input_size.observe(on_input_change, names='value')\n",
        "input_index.observe(on_input_change, names='value')\n",
        "\n",
        "add_button.on_click(add_row)\n",
        "delete_button.on_click(delete_row)\n",
        "update_button.on_click(update_row)\n",
        "save_button.on_click(save_row)\n",
        "read_button.on_click(read_row)\n",
        "\n",
        "# 显示表格和交互式按钮\n",
        "lock_buttons()\n",
        "show_table()"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "hMCNvrQSxnBI"
      },
      "outputs": [],
      "source": [
        "#@title ## 2.2、按照2.1的列表开始下载\n",
        "#@markdown ####（可选）只进行部分下载就填（复制名字按“与“字分割）：##\n",
        "mod_name = \"\" #@param {type: \"string\"}\n",
        "sd0=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"\n",
        "sd=sd0+'i'\n",
        "#@markdown ####（可选）是否跳过ckpt：##\n",
        "jump_ckpt = False # @param {type:'boolean'}\n",
        "\n",
        "import json\n",
        "import re\n",
        "import subprocess\n",
        "import shutil\n",
        "import os\n",
        "from concurrent.futures import ThreadPoolExecutor\n",
        "\n",
        "check_dir=f\"/content/{sd}/models/Stable-diffusion\"\n",
        "lora_dir=f\"/content/{sd}/models/Lora\"\n",
        "\n",
        "if 'ckpt_path' in globals() and ckpt_path!=\"\":\n",
        "  check_dir = f\"{ckpt_path}/\"\n",
        "\n",
        "def swap(arr,l,r):\n",
        "  temp=arr[l]\n",
        "  arr[l]=arr[r]\n",
        "  arr[r]=temp\n",
        "\n",
        "def checkRelateMod(oldMod, newMod):\n",
        "  try:\n",
        "    return oldMod and newMod and oldMod['downloadLink'] == newMod['downloadLink']\n",
        "  except Exception as e:\n",
        "    print(e)\n",
        "\n",
        "\n",
        "def sanitize_filename(filename):\n",
        "    # 定义不兼容字符的正则表达式模式\n",
        "    pattern = r'[\\\\/:*?\"<>|]'\n",
        "    # 使用下划线替换不兼容字符\n",
        "    sanitized_filename = re.sub(pattern, '_', filename)\n",
        "    return sanitized_filename\n",
        "\n",
        "def dealRelateMod(oldRenameMod, newRenameMod):\n",
        "  old_file_path = f\"{oldRenameMod['path']}/{oldRenameMod['name']}\"\n",
        "  #新模型的定义不带后缀名，所以需要加上处理，且在预处理情况下不知道路径那么只能沿用老模型的路径和名字\n",
        "  new_file_dir= (\n",
        "    newRenameMod['path']\n",
        "    if newRenameMod.get('path') and newRenameMod['path'] != \"\"\n",
        "    else oldRenameMod['path']\n",
        "    )\n",
        "  new_file_name=(\n",
        "      sanitize_filename(newRenameMod['name'])+os.path.splitext(oldRenameMod['name'])[1]\n",
        "      if newRenameMod['name'] and newRenameMod['name'] !=\"未命名\"\n",
        "      else oldRenameMod['name']\n",
        "    )\n",
        "  new_file_path = f\"{new_file_dir}/{new_file_name}\"\n",
        "  if old_file_path != new_file_path:\n",
        "    try:\n",
        "        if os.path.exists(old_file_path):\n",
        "            os.rename(old_file_path,new_file_path)\n",
        "            oldRenameMod['path']=new_file_dir\n",
        "            oldRenameMod['name']=new_file_name\n",
        "            print(f\"{old_file_path}更改为{new_file_path}成功\")\n",
        "        else:\n",
        "            print(f\"{old_file_path}不存在，重命名失败\")\n",
        "    except OSError as e:\n",
        "        print(f\"更改{old_file_path}时发生错误: {e}\")\n",
        "  else:\n",
        "    print(f\"{newRenameMod['name']}已存在且无任何改变，将忽视\")\n",
        "\n",
        "def deleteMod(oldDeMod):\n",
        "  file_path = f\"{oldDeMod['path']}/{oldDeMod['name']}\"\n",
        "  try:\n",
        "      if os.path.exists(file_path):\n",
        "          os.remove(file_path)\n",
        "          #oldCo的行也要进行同步删除\n",
        "          print(f\"{oldDeMod['name']}删除成功\")\n",
        "          del oldDeMod\n",
        "      else:\n",
        "          print(f\"{oldDeMod['name']}不存在，删除失败\")\n",
        "  except OSError as e:\n",
        "      print(f\"删除{oldDeMod['name']}时发生错误: {e}\")\n",
        "\n",
        "def LCS(oldCo, newCo):\n",
        "  oldStartIdx = 0   # 旧前指针\n",
        "  newStartIdx = 0   # 新前指针\n",
        "  oldEndIdx = len(oldCo) - 1  # 旧后指针\n",
        "  newEndIdx = len(newCo) - 1  # 新后指针\n",
        "  oldStartMod = oldCo[oldStartIdx]  # 旧前Mod\n",
        "  oldEndMod = oldCo[oldEndIdx]  # 旧后Mod\n",
        "  newStartMod = newCo[newStartIdx]  # 新前Mod\n",
        "  newEndMod = newCo[newEndIdx]  # 新后Mod\n",
        "  linkMap = {}\n",
        "  while oldStartIdx <= oldEndIdx and newStartIdx <= newEndIdx:\n",
        "    # 新前旧前\n",
        "    if checkRelateMod(oldStartMod, newStartMod):\n",
        "      dealRelateMod(oldStartMod, newStartMod)\n",
        "      oldStartIdx += 1\n",
        "      newStartIdx += 1\n",
        "      if newStartIdx > newEndIdx or oldStartIdx > oldEndIdx:\n",
        "        break\n",
        "      oldStartMod = oldCo[oldStartIdx]\n",
        "      newStartMod = newCo[newStartIdx]\n",
        "    # 新后旧后\n",
        "    elif checkRelateMod(oldEndMod, newEndMod):\n",
        "      dealRelateMod(oldEndMod, newEndMod)\n",
        "      oldEndIdx -= 1\n",
        "      newEndIdx -= 1\n",
        "      if newStartIdx > newEndIdx or oldStartIdx > oldEndIdx:\n",
        "        break\n",
        "      oldEndMod = oldCo[oldEndIdx]\n",
        "      newEndMod = newCo[newEndIdx]\n",
        "    # 新后旧前\n",
        "    elif checkRelateMod(oldStartMod, newEndMod):\n",
        "      dealRelateMod(oldStartMod, newEndMod)\n",
        "      oldStartIdx += 1\n",
        "      newEndIdx -= 1\n",
        "      if newStartIdx > newEndIdx or oldStartIdx > oldEndIdx:\n",
        "        break\n",
        "      oldStartMod = oldCo[oldStartIdx]\n",
        "      newEndMod = newCo[newEndIdx]\n",
        "    # 新前旧后\n",
        "    elif checkRelateMod(oldEndMod, newStartMod):\n",
        "      dealRelateMod(oldEndMod, newStartMod)\n",
        "      oldEndIdx -= 1\n",
        "      newStartIdx += 1\n",
        "      if newStartIdx > newEndIdx or oldStartIdx > oldEndIdx:\n",
        "        break\n",
        "      oldEndMod = oldCo[oldEndIdx]\n",
        "      newStartMod = newCo[newStartIdx]\n",
        "    # 四种均未找到\n",
        "    else:\n",
        "      # if not linkMap:\n",
        "      linkMap = {}\n",
        "    # 从 oldStartIdx 开始，到oldEndIdx结束，创建linkMap映射对象\n",
        "      for i in range(oldStartIdx, oldEndIdx+1):\n",
        "        downloadLink = oldCo[i]['downloadLink']\n",
        "        if downloadLink is not None:\n",
        "          linkMap[downloadLink] = i\n",
        "      idxInOld = linkMap.get(newStartMod['downloadLink'], None)\n",
        "      if idxInOld is None:\n",
        "        print(f\"{newStartMod['name']}不存在于旧mod,{newStartMod['downloadLink']}将添加到下载任务队列\")\n",
        "        content.append(newStartMod)\n",
        "\n",
        "      else:\n",
        "        dealRelateMod(oldCo[idxInOld], newStartMod)\n",
        "        swap(oldCo,idxInOld,oldEndIdx)\n",
        "        oldEndIdx-=1\n",
        "      # 指针下移，移动新的头\n",
        "      newStartIdx += 1\n",
        "      if newStartIdx > newEndIdx:\n",
        "          break\n",
        "      newStartMod = newCo[newStartIdx]\n",
        "      oldEndMod=oldCo[oldEndIdx]\n",
        "      # oldStartIdx += 1\n",
        "  print('new',newStartIdx,newEndIdx)\n",
        "  print('old',oldStartIdx,oldEndIdx)\n",
        "  # new这里还有剩余Mod没有处理\n",
        "  if newStartIdx <= newEndIdx:\n",
        "    for i in range(newStartIdx, newEndIdx + 1):\n",
        "      print(f\"{newCo[i]['name']}为新且不重复mod，将添加到下载任务队列\")\n",
        "      content.append(newCo[i])\n",
        "\n",
        "  elif oldStartIdx <= oldEndIdx:\n",
        "    for i in range(oldStartIdx, oldEndIdx + 1):\n",
        "      if oldCo[i]:\n",
        "        print(f\"{oldCo[i]['name']}为旧不重复mod，将删除\")\n",
        "        deleteMod(oldCo[i])\n",
        "\n",
        "# LCS(oldArr,oldArr)\n",
        "with open(mod_json_dir, 'r') as modelFile:\n",
        "  preContent=json.loads(modelFile.read())\n",
        "\n",
        "#更新阶段\n",
        "if 'oldCo' in globals() and len(oldCo)>0:\n",
        "  content=[]\n",
        "  print(oldCo)\n",
        "  print(preContent)\n",
        "  LCS(oldCo, preContent)\n",
        "  print(content)\n",
        "#初始化阶段\n",
        "else:\n",
        "  oldCo=[]\n",
        "  content=preContent\n",
        "\n",
        "\n",
        "#部分下载的情况\n",
        "selected_mods = [x.strip() for x in mod_name.split('与') if x.strip()]\n",
        "if mod_name in selected_mods:\n",
        "  selected_mods = []\n",
        "\n",
        "def get_file_size(file_path):\n",
        "    try:\n",
        "        file_size_bytes = os.path.getsize(file_path)\n",
        "        for unit in ['B', 'KB', 'MB', 'GB', 'TB']:\n",
        "            if file_size_bytes < 1024.0:\n",
        "                break\n",
        "            file_size_bytes /= 1024.0\n",
        "        return f\"{file_size_bytes:.2f} {unit}\"\n",
        "    except OSError as e:\n",
        "        print(f\"Error: {e}\")\n",
        "        return None\n",
        "\n",
        "def get_civitai_file(str):\n",
        "    match = re.search(r'[^/]+$', str.decode())\n",
        "    if match:\n",
        "      return match.group(0).split('\\n')[0]\n",
        "    else:\n",
        "      print(\"无法从输出中提取文件名\")\n",
        "\n",
        "def move_model(source_path, target_path):\n",
        "    try:\n",
        "        shutil.move(source_path, target_path)\n",
        "    except OSError as e:\n",
        "        print(f\"Error: {e}\")\n",
        "\n",
        "def download_file(item):\n",
        "  # for item in content:\n",
        "    if selected_mods and item['name'] not in selected_mods:\n",
        "        return\n",
        "    \n",
        "    if jump_ckpt and item['type'] == 'checkpoint':\n",
        "        return\n",
        "    \n",
        "    download_url=item['downloadLink']\n",
        "    #huggingface\n",
        "    match = re.search(r'/([^/]*)$', download_url)\n",
        "    file_name = match.group(1)\n",
        "    #开始下载\n",
        "    cmd = f\"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {lora_dir}\" if '.' not in match.group(1) else f\"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {lora_dir} -o {file_name}\"\n",
        "    result = subprocess.run(cmd, shell=True, stderr=subprocess.PIPE, stdout=subprocess.PIPE)\n",
        "    if result.returncode == 0:\n",
        "      # 判断是civitai还是hugging face还是自定义\n",
        "      #civitai\n",
        "      if '.' not in file_name:\n",
        "        file_name =get_civitai_file(result.stdout)\n",
        "      #如果定义了名字则重命名\n",
        "      if item['name']!='未命名':\n",
        "        #需要进行文件名的兼容处理\n",
        "        temp_name=sanitize_filename(item['name'])+os.path.splitext(file_name)[1]\n",
        "        os.rename(f\"{lora_dir}/{file_name}\",f\"{lora_dir}/{temp_name}\")\n",
        "        final_name =temp_name\n",
        "      else:\n",
        "        final_name =file_name\n",
        "      source_path = f\"{lora_dir}/{final_name}\"\n",
        "      file_size = get_file_size(source_path)\n",
        "      #如果path定义了则进行移动\n",
        "      if item.get('path') and item['path'] != \"\":\n",
        "          target_path = item['path']\n",
        "          move_model(source_path, target_path)\n",
        "          item['path']=target_path\n",
        "          print(file_name+'已下载，重命名为：'+final_name)\n",
        "          print('移动--',final_name,f'到{target_path}')\n",
        "      #如果为checkpoint则进行移动\n",
        "      elif file_size and 'GB'in file_size:\n",
        "          move_model(source_path, check_dir)\n",
        "          item['path']=check_dir\n",
        "          print(file_name+'已下载，重命名为：'+final_name)\n",
        "          print('移动checkpoint--',final_name,f'到{check_dir}文件夹')\n",
        "      #如果为lora则直接调用source_path\n",
        "      else:\n",
        "          item['path']=lora_dir\n",
        "          print(file_name+'已下载，重命名为：'+final_name)\n",
        "      #改变item['name']用于最小化更新的重命名判断\n",
        "      item['name']=final_name\n",
        "      oldCo.append(item)\n",
        "    else:\n",
        "      print(f\"{item['name']}下载失败,请检查{item['downloadLink']}\")\n",
        "\n",
        "# download_file(item)\n",
        "# 使用线程池进行下载\n",
        "with ThreadPoolExecutor(max_workers=5) as executor:\n",
        "    for item in content:\n",
        "        executor.submit(download_file, item)\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "zwkiS9z3X2Gz"
      },
      "outputs": [],
      "source": [
        "#@title ## 3、初始化/重置到推荐配置\n",
        "#@markdown ####是否允许在云盘创建图片文件夹且将生成的图片导入到云盘：##\n",
        "is_cloud_output = False  # @param {type:'boolean'}\n",
        "#@markdown ####生成图片自动保存到本地设备：##\n",
        "is_auto_download = False  # @param {type:'boolean'}\n",
        "#@markdown ####使用你云盘里的config.json：##\n",
        "is_config_json = False  # @param {type:'boolean'}\n",
        "sd0=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"\n",
        "sd=sd0+'i'\n",
        "wi='w'+'e'+'b'+'u'+'i'\n",
        "str_temp2='sd'+'-'+wi\n",
        "\n",
        "import json\n",
        "import re\n",
        "import os\n",
        "\n",
        "#自动查找config.json文件\n",
        "def find_config_json(root_path,config_json_name):\n",
        "    for file_name in os.listdir(root_path):\n",
        "        file_path = os.path.join(root_path, file_name)\n",
        "        if os.path.isdir(file_path):\n",
        "            res=find_model_json(file_path,config_json_name)\n",
        "            if res is not None:\n",
        "              return res\n",
        "        elif file_name == config_json_name:\n",
        "            print(\"找到config.json文件：\", file_path)\n",
        "            return file_path\n",
        "\n",
        "def rec_config_json():\n",
        "  json_content = {\n",
        "      \"samples_save\": True,\n",
        "      \"samples_format\": \"png\",\n",
        "      \"samples_filename_pattern\": \"\",\n",
        "      \"save_images_add_number\": True,\n",
        "      \"grid_save\": True,\n",
        "      \"grid_format\": \"png\",\n",
        "      \"grid_extended_filename\": False,\n",
        "      \"grid_only_if_multiple\": True,\n",
        "      \"grid_prevent_empty_spots\": False,\n",
        "      \"n_rows\": -1,\n",
        "      \"enable_pnginfo\": True,\n",
        "      \"save_txt\": False,\n",
        "      \"save_images_before_face_restoration\": False,\n",
        "      \"save_images_before_highres_fix\": False,\n",
        "      \"save_images_before_color_correction\": False,\n",
        "      \"jpeg_quality\": 80,\n",
        "      \"export_for_4chan\": True,\n",
        "      \"use_original_name_batch\": True,\n",
        "      \"use_upscaler_name_as_suffix\": False,\n",
        "      \"save_selected_only\": True,\n",
        "      \"do_not_add_watermark\": False,\n",
        "      \"temp_dir\": \"\",\n",
        "      \"clean_temp_dir_at_start\": False,\n",
        "      \"outdir_samples\": \"\",\n",
        "      \"outdir_txt2img_samples\": \"outputs/txt2img-images\",\n",
        "      \"outdir_img2img_samples\": \"outputs/img2img-images\",\n",
        "      \"outdir_extras_samples\": \"outputs/extras-images\",\n",
        "      \"outdir_grids\": \"\",\n",
        "      \"outdir_txt2img_grids\": \"outputs/txt2img-grids\",\n",
        "      \"outdir_img2img_grids\": \"outputs/img2img-grids\",\n",
        "      \"outdir_save\": \"log/images\",\n",
        "      \"save_to_dirs\": False,\n",
        "      \"grid_save_to_dirs\": False,\n",
        "      \"use_save_to_dirs_for_ui\": False,\n",
        "      \"directories_filename_pattern\": \"[date]\",\n",
        "      \"directories_max_prompt_words\": 8,\n",
        "      \"ESRGAN_tile\": 192,\n",
        "      \"ESRGAN_tile_overlap\": 8,\n",
        "      \"realesrgan_enabled_models\": [\n",
        "          \"R-ESRGAN 4x+\",\n",
        "          \"R-ESRGAN 4x+ Anime6B\"\n",
        "      ],\n",
        "      \"upscaler_for_img2img\": None,\n",
        "      \"face_restoration_model\": None,\n",
        "      \"code_former_weight\": 0.5,\n",
        "      \"face_restoration_unload\": False,\n",
        "      \"show_warnings\": False,\n",
        "      \"memmon_poll_rate\": 8,\n",
        "      \"samples_log_stdout\": False,\n",
        "      \"multiple_tqdm\": True,\n",
        "      \"print_hypernet_extra\": False,\n",
        "      \"unload_models_when_training\": True,\n",
        "      \"pin_memory\": False,\n",
        "      \"save_optimizer_state\": False,\n",
        "      \"save_training_settings_to_txt\": True,\n",
        "      \"dataset_filename_word_regex\": \"\",\n",
        "      \"dataset_filename_join_string\": \" \",\n",
        "      \"training_image_repeats_per_epoch\": 1,\n",
        "      \"training_write_csv_every\": 500,\n",
        "      \"training_xattention_optimizations\": False,\n",
        "      \"training_enable_tensorboard\": False,\n",
        "      \"training_tensorboard_save_images\": False,\n",
        "      \"training_tensorboard_flush_every\": 120,\n",
        "      \"sd_model_checkpoint\": \"\",\n",
        "      \"sd_checkpoint_cache\": 0,\n",
        "      \"sd_vae_checkpoint_cache\": 0,\n",
        "      \"sd_vae\": \"Automatic\",\n",
        "      \"sd_vae_as_default\": False,\n",
        "      \"inpainting_mask_weight\": 1.0,\n",
        "      \"initial_noise_multiplier\": 1.0,\n",
        "      \"img2img_color_correction\": False,\n",
        "      \"img2img_fix_steps\": False,\n",
        "      \"img2img_background_color\": \"#ffffff\",\n",
        "      \"enable_quantization\": False,\n",
        "      \"enable_emphasis\": True,\n",
        "      \"enable_batch_seeds\": True,\n",
        "      \"comma_padding_backtrack\": 20,\n",
        "      \"CLIP_stop_at_last_layers\": 2,\n",
        "      \"upcast_attn\": False,\n",
        "      \"use_old_emphasis_implementation\": False,\n",
        "      \"use_old_karras_scheduler_sigmas\": False,\n",
        "      \"use_old_hires_fix_width_height\": False,\n",
        "      \"interrogate_keep_models_in_memory\": False,\n",
        "      \"interrogate_return_ranks\": False,\n",
        "      \"interrogate_clip_num_beams\": 1,\n",
        "      \"interrogate_clip_min_length\": 24,\n",
        "      \"interrogate_clip_max_length\": 48,\n",
        "      \"interrogate_clip_dict_limit\": 1500,\n",
        "      \"interrogate_clip_skip_categories\": [],\n",
        "      \"interrogate_deepbooru_score_threshold\": 0.7,\n",
        "      \"deepbooru_sort_alpha\": False,\n",
        "      \"deepbooru_use_spaces\": False,\n",
        "      \"deepbooru_escape\": True,\n",
        "      \"deepbooru_filter_tags\": \"\",\n",
        "      \"extra_networks_default_view\": \"cards\",\n",
        "      \"extra_networks_default_multiplier\": 1.0,\n",
        "      \"sd_hypernetwork\": \"None\",\n",
        "      \"return_grid\": False,\n",
        "      \"do_not_show_images\": False,\n",
        "      \"add_model_hash_to_info\": True,\n",
        "      \"add_model_name_to_info\": True,\n",
        "      \"disable_weights_auto_swap\": True,\n",
        "      \"send_seed\": True,\n",
        "      \"send_size\": True,\n",
        "      \"font\": \"\",\n",
        "      \"js_modal_lightbox\": True,\n",
        "      \"js_modal_lightbox_initially_zoomed\": True,\n",
        "      \"show_progress_in_title\": True,\n",
        "      \"samplers_in_dropdown\": False,\n",
        "      \"dimensions_and_batch_together\": True,\n",
        "      \"keyedit_precision_attention\": 0.1,\n",
        "      \"keyedit_precision_extra\": 0.05,\n",
        "      \"quicksettings\": \"sd_model_checkpoint, sd_vae, CLIP_stop_at_last_layers, use_old_karras_scheduler_sigmas, always_discard_next_to_last_sigma\",\n",
        "      \"ui_reorder\": \"inpaint, sampler, checkboxes, hires_fix, dimensions, cfg, seed, batch, override_settings, scripts\",\n",
        "      \"ui_extra_networks_tab_reorder\": \"\",\n",
        "      \"localization\": \"zh-Hans (Stable)\",\n",
        "      \"show_progressbar\": True,\n",
        "      \"live_previews_enable\": True,\n",
        "      \"show_progress_grid\": True,\n",
        "      \"show_progress_every_n_steps\": 20,\n",
        "      \"show_progress_type\": \"Approx NN\",\n",
        "      \"live_preview_content\": \"Prompt\",\n",
        "      \"live_preview_refresh_period\": 1000,\n",
        "      \"hide_samplers\": [],\n",
        "      \"eta_ddim\": 0.0,\n",
        "      \"eta_ancestral\": 1.0,\n",
        "      \"ddim_discretize\": \"uniform\",\n",
        "      \"s_churn\": 0.0,\n",
        "      \"s_tmin\": 0.0,\n",
        "      \"s_noise\": 1.0,\n",
        "      \"eta_noise_seed_delta\": 31337,\n",
        "      \"always_discard_next_to_last_sigma\": False,\n",
        "      \"postprocessing_enable_in_main_ui\": [],\n",
        "      \"postprocessing_operation_order\": [],\n",
        "      \"upscaling_max_images_in_cache\": 5,\n",
        "      \"disabled_extensions\": [],\n",
        "      \"sd_checkpoint_hash\": \"7af57400eb7303877ec35e5b9e03fc29802c44066828165dc3a20b973c439428\",\n",
        "      \"ldsr_steps\": 100,\n",
        "      \"ldsr_cached\": False,\n",
        "      \"SWIN_tile\": 192,\n",
        "      \"SWIN_tile_overlap\": 8,\n",
        "      \"sd_lora\": \"None\",\n",
        "      \"lora_apply_to_outputs\": False,\n",
        "      \"tac_tagFile\": \"danbooru.csv\",\n",
        "      \"tac_active\": True,\n",
        "      \"tac_activeIn.txt2img\": True,\n",
        "      \"tac_activeIn.img2img\": True,\n",
        "      \"tac_activeIn.negativePrompts\": True,\n",
        "      \"tac_activeIn.thirdParty\": True,\n",
        "      \"tac_activeIn.modelList\": \"\",\n",
        "      \"tac_activeIn.modelListMode\": \"Blacklist\",\n",
        "      \"tac_maxResults\": 15.0,\n",
        "      \"tac_showAllResults\": False,\n",
        "      \"tac_resultStepLength\": 100.0,\n",
        "      \"tac_delayTime\": 100.0,\n",
        "      \"tac_useWildcards\": True,\n",
        "      \"tac_useEmbeddings\": True,\n",
        "      \"tac_useHypernetworks\": True,\n",
        "      \"tac_useLoras\": True,\n",
        "      \"tac_showWikiLinks\": False,\n",
        "      \"tac_replaceUnderscores\": True,\n",
        "      \"tac_escapeParentheses\": True,\n",
        "      \"tac_appendComma\": True,\n",
        "      \"tac_alias.searchByAlias\": True,\n",
        "      \"tac_alias.onlyShowAlias\": False,\n",
        "      \"tac_translation.translationFile\": \"CN.csv\",\n",
        "      \"tac_translation.oldFormat\": True,\n",
        "      \"tac_translation.searchByTranslation\": True,\n",
        "      \"tac_extra.extraFile\": \"extra-quality-tags.csv\",\n",
        "      \"tac_extra.addMode\": \"Insert before\",\n",
        "      \"additional_networks_extra_lora_path\": \"\",\n",
        "      \"additional_networks_sort_models_by\": \"name\",\n",
        "      \"additional_networks_reverse_sort_order\": False,\n",
        "      \"additional_networks_model_name_filter\": \"\",\n",
        "      \"additional_networks_xy_grid_model_metadata\": \"\",\n",
        "      \"additional_networks_hash_thread_count\": 1.0,\n",
        "      \"additional_networks_back_up_model_when_saving\": True,\n",
        "      \"additional_networks_show_only_safetensors\": False,\n",
        "      \"additional_networks_show_only_models_with_metadata\": \"disabled\",\n",
        "      \"additional_networks_max_top_tags\": 20.0,\n",
        "      \"additional_networks_max_dataset_folders\": 20.0,\n",
        "      \"images_history_preload\": False,\n",
        "      \"images_record_paths\": True,\n",
        "      \"images_delete_message\": True,\n",
        "      \"images_history_page_columns\": 6.0,\n",
        "      \"images_history_page_rows\": 6.0,\n",
        "      \"images_history_pages_perload\": 20.0,\n",
        "      \"img_downscale_threshold\": 4.0,\n",
        "      \"target_side_length\": 4000.0,\n",
        "      \"no_dpmpp_sde_batch_determinism\": False,\n",
        "      \"control_net_max_models_num\": 5,\n",
        "  }\n",
        "  if is_cloud_output:\n",
        "    !mkdir -p /content/{sd}/\n",
        "    json_content[\"outdir_txt2img_samples\"]=\"/content/drive/MyDrive/outputs/txt2img-images\"\n",
        "    json_content[\"outdir_img2img_samples\"]=\"/content/drive/MyDrive/outputs/img2img-images\"\n",
        "    json_content[\"outdir_extras_samples\"]=\"/content/drive/MyDrive/outputs/extras-images\"\n",
        "    json_content[\"outdir_txt2img_grids\"]=\"/content/drive/MyDrive/outputs/txt2img-grids\"\n",
        "    json_content[\"outdir_img2img_grids\"]=\"/content/drive/MyDrive/outputs/img2img-grids\"\n",
        "  with open(f'/content/{sd}/config.json', 'w') as configFile:\n",
        "    json.dump(json_content, configFile, ensure_ascii=False, indent=4)\n",
        "\n",
        "if is_config_json==False:\n",
        "  rec_config_json()\n",
        "else:\n",
        "  res_config_json=find_config_json('/content/drive/','config.json')\n",
        "  print(res_config_json)\n",
        "  !cp {res_config_json} /content/{sd}/config.json\n",
        "#图片自动下载脚本\n",
        "if is_auto_download:\n",
        "  !wget -O /content/{sd}/javascript/png_auto_download.js https://github.com/ennnnny/sd_colab/raw/main/png_auto_download.js\n",
        "else:\n",
        "  if os.path.exists(f'/content/{sd}/javascript/png_auto_download.js'):\n",
        "    os.remove(f'/content/{sd}/javascript/png_auto_download.js')\n",
        "\n",
        "\n",
        "#anapnoe版本\n",
        "if ssdd_ui==\"anapnoe手机端完美适配\":\n",
        "  #png_info脚本\n",
        "  !wget -O /content/{sd}/javascript/PNG_info_web.js https://github.com/ennnnny/sd_colab/raw/main/PNG_info_web.js\n",
        "  with open(f'/content/{sd}/modules/ui.py', 'r') as readFile:\n",
        "    content = readFile.read()\n",
        "  content=content.replace('''                for tabname, button in buttons.items():\n",
        "                    parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(\n",
        "                        paste_button=button, tabname=tabname, source_text_component=generation_info, source_image_component=image,\n",
        "                    ))\n",
        "\n",
        "        image.change(\n",
        "            fn=wrap_gradio_call(modules.extras.run_pnginfo),\n",
        "            inputs=[image],\n",
        "            outputs=[html, generation_info, html2],\n",
        "        )''','')\n",
        "  content=content.replace('''[\"txt2img\", \"img2img\", \"inpaint\", \"extras\"]''','[\"txt2img\", \"img2img\", \"inpaint\", \"tagger图生文\"]')\n",
        "  \n",
        "  with open(f'/content/{sd}/modules/ui.py', 'w') as writeFile:\n",
        "      writeFile.write(content)\n",
        "  #png_info脚本兼容\n",
        "  with open(f'/content/{sd}/javascript/PNG_info_web.js', 'r') as readFile:\n",
        "    content = readFile.read()\n",
        "  content=content.replace('''querySelector(\"#tab_pnginfo > div > div > div:nth-child(2) > div:nth-child(3)\")''','querySelector(\"#tab_pnginfo > div > div>div>div:nth-child(4)\")')\n",
        "  with open(f'/content/{sd}/javascript/PNG_info_web.js', 'w') as writeFile:\n",
        "      writeFile.write(content)\n",
        "\n",
        "#automatic111版本\n",
        "else:\n",
        "  #手机平板\n",
        "  css_content = '''\n",
        "  @media screen and (max-width: 600px) {\n",
        "    .gradio-slider input[type=\"range\"]{\n",
        "      display: none;\n",
        "    }\n",
        "    .gradio-slider input[type=\"number\"]{\n",
        "      width: 18em;\n",
        "  }\n",
        "  }\n",
        "  '''\n",
        "  with open(f'/content/{sd}/style.css', 'a') as cssFile:\n",
        "      cssFile.write(css_content)\n",
        "  #png_info脚本\n",
        "  !wget -O /content/{sd}/javascript/PNG_info_web.js https://github.com/ennnnny/sd_colab/raw/main/PNG_info_web.js\n",
        "  #拦截png_info\n",
        "  with open(f'/content/{sd}/modules/ui.py', 'r') as readFile:\n",
        "    content = readFile.read()\n",
        "  content=content.replace('''                for tabname, button in buttons.items():\n",
        "                    parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(\n",
        "                        paste_button=button, tabname=tabname, source_text_component=generation_info, source_image_component=image,\n",
        "                    ))\n",
        "\n",
        "        image.change(\n",
        "            fn=wrap_gradio_call(modules.extras.run_pnginfo),\n",
        "            inputs=[image],\n",
        "            outputs=[html, generation_info, html2],\n",
        "        )''','')\n",
        "  content=content.replace('''[\"txt2img\", \"img2img\", \"inpaint\", \"extras\"]''','[\"txt2img\", \"img2img\", \"inpaint\", \"tagger图生文\"]')\n",
        "  with open(f'/content/{sd}/modules/ui.py', 'w') as writeFile:\n",
        "    writeFile.write(content)\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "2kyb_9_rRFpA"
      },
      "outputs": [],
      "source": [
        "#@title # **4、运行/重启**\n",
        "sd0=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"\n",
        "sd=sd0+'i'\n",
        "%cd /content/{sd}\n",
        "#@markdown ####全精度/半精度(推荐)启动：##\n",
        "is_full_precision = False  # @param {type:'boolean'}\n",
        "#@markdown ####主题切换为暗配色：##\n",
        "is_black = True  # @param {type:'boolean'}\n",
        "#@markdown ####(可选)获取[ngrok](https://dashboard.ngrok.com/get-started/your-authtoken)的token进行免费网络加速：##\n",
        "ngrok_auth=\"\"  #@param {type:\"string\"}\n",
        "ngrok_region= \"auto\" #@param [\"auto\", \"us\", \"eu\", \"au\", \"ap\", \"sa\", \"jp\", \"in\"]\n",
        "is_cloudflared = False  # @param {type:'boolean'}\n",
        "is_localhostrun = False  # @param {type:'boolean'}\n",
        "is_remotemoe = False  # @param {type:'boolean'}\n",
        "is_googleusercontent = False  # @param {type:'boolean'}\n",
        "is_jprq = False # @param {type:'boolean'}\n",
        "jprq_token = \"\"  #@param {type:\"string\"}\n",
        "is_bore = False # @param {type:'boolean'}\n",
        "bore_url = \"\"  #@param {type:\"string\"}\n",
        "\n",
        "base_str=\"--share --lowram --disable-safe-unpickle  --disable-console-progressbars --xformers --enable-insecure-extension-access\"\n",
        "if is_black:\n",
        "  base_str+=f\" --theme=dark\"\n",
        "else:\n",
        "  base_str+=f\" --theme=light\"\n",
        "if ngrok_auth:\n",
        "  base_str+=f\"  --ngrok={ngrok_auth} --ngrok-region={ngrok_region}\"\n",
        "if is_cloudflared:\n",
        "  base_str+=f\"  --cloudflared\"\n",
        "if is_localhostrun:\n",
        "  base_str+=f\"  --localhostrun\"\n",
        "if is_remotemoe:\n",
        "  base_str+=f\"  --remotemoe\"\n",
        "if is_googleusercontent:\n",
        "  base_str+=f\"  --googleusercontent\"\n",
        "if is_jprq and jprq_token != \"\":\n",
        "  base_str+=f\"  --jprq `{jprq_token}`\"\n",
        "if is_bore:\n",
        "  base_str+=f\"  --bore\"\n",
        "  if bore_url != \"\":\n",
        "    base_str+=f\" --bore_url `{bore_url}`\"\n",
        "\n",
        "if 'ckpt_path' in globals() and ckpt_path!=\"\":\n",
        "  base_str+=f'  --ckpt-dir \"{ckpt_path}\"'\n",
        "\n",
        "full_precision_str=base_str + f\"  --precision full --no-half --no-half-vae --opt-sub-quad-attention --opt-channelslast --api\"\n",
        "half_precision_str=base_str + f\"  --opt-sub-quad-attention --opt-channelslast --api\"\n",
        "\n",
        "if is_full_precision:\n",
        "  !python launch.py {full_precision_str} #（解决精度不足但速度不够）\n",
        "else:\n",
        "  !python launch.py {half_precision_str} #半精度（速度提升1倍以上，但可能出现精度不足问题）"
      ]
    },
    {
      "attachments": {},
      "cell_type": "markdown",
      "metadata": {
        "id": "-tq6a-O-STNt"
      },
      "source": [
        "# **推荐模型文档下载和删除sd**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@title ## 新人推荐mod集合\n",
        "#@markdown ####将下载json到云盘：##\n",
        "mod_json_name = \"model.json\" #@param [\"model.json\", \"Anime.json\", \"anime2.json\", \"真人.json\", \"realistic.json\", \"design.json\", \"gameCG.json\"]\n",
        "\n",
        "mod_download_url={\n",
        "   \"model.json\": \"https://raw.githubusercontent.com/ennnnny/sd_colab/self/model.json\",\n",
        "   \"Anime.json\": \"https://huggingface.co/datasets/daasd/model_json/resolve/main/Anime.json\",\n",
        "   \"anime2.json\": \"https://raw.githubusercontent.com/ennnnny/sd_colab/self/anime2.json\",\n",
        "   \"真人.json\": \"https://huggingface.co/datasets/daasd/model_json/resolve/main/%E7%9C%9F%E4%BA%BA.json\",\n",
        "   \"realistic.json\": \"https://raw.githubusercontent.com/ennnnny/sd_colab/self/realistic.json\",\n",
        "   \"design.json\": \"https://huggingface.co/datasets/daasd/model_json/resolve/main/design.json\",\n",
        "   \"gameCG.json\": \"https://huggingface.co/datasets/daasd/model_json/resolve/main/gameCG.json\"\n",
        "}\n",
        "\n",
        "if is_load_pan==\"允许\":\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {mod_download_url[mod_json_name]} -d /content/drive/MyDrive/\n",
        "  mod_json_dir=f'/content/drive/MyDrive/{mod_json_name}'\n",
        "else:\n",
        "  !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {mod_download_url[mod_json_name]} -d /content/\n",
        "  mod_json_dir=f'/content/{mod_json_name}'"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "y2gQLc3h2LHg"
      },
      "outputs": [],
      "source": [
        "#@title ## 删除临时硬盘的SD，重新安装和部署（别乱点）\n",
        "sd=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"+\"i\"\n",
        "import os\n",
        "import shutil\n",
        "import ipywidgets as widgets\n",
        "import time\n",
        "from IPython.display import display\n",
        "from google.colab import output\n",
        "\n",
        "def delete_lora_folder(b):\n",
        "    folder_path = f'/content/{sd}'\n",
        "    if os.path.exists(folder_path):\n",
        "        shutil.rmtree(folder_path)\n",
        "        print(\"已成功删除，请重新安装\")\n",
        "        time.sleep(5)\n",
        "        output.clear()\n",
        "    else:\n",
        "        print(\"已取消删除\")\n",
        "        time.sleep(5)\n",
        "        output.clear()\n",
        "\n",
        "def on_button_click(button):\n",
        "    delete_lora_folder(button)\n",
        "\n",
        "button = widgets.Button(description=\"确定要删除吗?\")\n",
        "button.on_click(on_button_click)\n",
        "display(button)\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "watg5f_gTNDI"
      },
      "outputs": [],
      "source": [
        "#@title ## 其他自定义下载\n",
        "#@markdown ####下载地址：##\n",
        "download_url = \"\" #@param {type: \"string\"}\n",
        "#@markdown ####存放目录：##\n",
        "download_dir = \"\" #@param {type: \"string\"}\n",
        "#@markdown ####文件名：##\n",
        "download_name = \"\" #@param {type: \"string\"}\n",
        "\n",
        "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {download_dir} -o {download_name} "
      ]
    },
    {
      "attachments": {},
      "cell_type": "markdown",
      "metadata": {},
      "source": [
        "---\n",
        "# **有用的程序 (可选)**\n",
        "\n",
        "*可能有用的东西*"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@markdown ## Colab Mobile Keep Alive \n",
        "#@markdown Run this cell to keep the tab alive in mobile (before running the start cell)\n",
        "\n",
        "%%html\n",
        "<b>Press play on the music player to keep the tab alive before running the start cell (Uses only 13MB of data)</b><br/>\n",
        "<audio src=\"https://raw.githubusercontent.com/KoboldAI/KoboldAI-Client/main/colab/silence.m4a\" controls>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@markdown ## Mass download outputs\n",
        "#@markdown Zip all image outputs and downloads them\n",
        "from google.colab import files\n",
        "import os\n",
        "%cd /content\n",
        "\n",
        "sd2=\"s\"+\"t\"+\"a\"+\"b\"+\"l\"+\"e\"+\"-\"+\"d\"+\"i\"+\"f\"+\"f\"+\"u\"+\"s\"+\"i\"+\"o\"+\"n\"+\"-\"+\"w\"+\"e\"+\"b\"+\"u\"+\"i\"\n",
        "c_path=\"/content/{sd2}/outputs\"\n",
        "if is_load_pan==\"允许\" and is_cloud_output:\n",
        "  c_path=\"/content/drive/MyDrive/outputs\"\n",
        "  \n",
        "if os.path.exists(\"/content/outputs.zip\"):\n",
        "  !rm -f /content/outputs.zip\n",
        "\n",
        "!cd {c_path} && zip -r /content/outputs.zip .\n",
        "files.download('/content/outputs.zip')"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {},
      "source": [
        "## XTerm Terminal\n",
        "*Spins up a terminal*"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@title Install XTerm\n",
        "!pip install colab-xterm\n",
        "%load_ext colabxterm"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "#@title Run XTerm (Terminal)\n",
        "%xterm"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "collapsed_sections": [
        "2EmivlLQZLRz",
        "6hHNXtOEiKJy",
        "-tq6a-O-STNt",
        "-DNtbo2FjPl0",
        "wv3agCiaApo6"
      ],
      "include_colab_link": true,
      "private_outputs": true,
      "provenance": []
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}