{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "source": [
        "#@markdown #下载webui程序到云盘\n",
        "import os,sys,time\n",
        "from IPython.utils import capture\n",
        "from IPython.display import clear_output\n",
        "SD_Folder_Name='sd'\n",
        "sd_dir=\"/content/gdrive/MyDrive/\"+SD_Folder_Name\n",
        "webui_dir=sd_dir+\"/stable-diffusion-\"+blasphemy+\"\"\n",
        "import base64\n",
        "t1=base64.b64decode((\"YzNSaFlteGxMV1JwWm1aMWMybHZiaTEzWldKMWFRPT0=\").encode('ascii')).decode('ascii')\n",
        "blasphemy0=base64.b64decode((t1).encode('ascii')).decode('ascii')\n",
        "#@markdown ！！如果之前版本运行有问题，先在谷歌云盘里删除SD文件夹并清空回收站再尝试\n",
        "\n",
        "#@markdown 勾选将下载NovelAi hypernetwork文件（下载速度可能很慢，新手可以不下载）：\n",
        "Download_Hypernetworks=False #@param {type:\"boolean\"}\n",
        "#@markdown 使用NovelAi参数（建议使用）\n",
        "UseConfig=False #@param {type:\"boolean\"}\n",
        "#@markdown 使用默认权重文件VAE\n",
        "VAEfile=True #@param {type:\"boolean\"}\n",
        "%cd $sd_dir\n",
        "!git clone -q --branch master https://github.com/AUTOMATIC1111/$blasphemy0\n",
        "!mkdir -p $webui_dir/models/Stable-diffusion $webui_dir/models/hypernetworks $webui_dir/models/deepbooru\n",
        "%cd $webui_dir/models/Stable-diffusion/\n",
        "if UseConfig:\n",
        "  !wget -O $webui_dir/config.json https://raw.githubusercontent.com/PtO2/stablediffusion_$blasphemy_configs/main/config.json\n",
        "  !wget -O $webui_dir/ui-config.json https://raw.githubusercontent.com/PtO2/stablediffusion_$blasphemy_configs/main/ui-config.json\n",
        "# Install VAE Weights (optional)\n",
        "if VAEfile:\n",
        "  !wget -O $sd_dir/animevae.pt https://cloudflare-ipfs.com/ipfs/bafybeiccldswdd3wvg57jhclcq53lvsc6gizasiblwayvhlv6eq4wow7wu/animevae.pt\n",
        "%cd /content\n",
        "# Install hypernetwork （optional)\n",
        "if Download_Hypernetworks:\n",
        "  !curl -L https://cloudflare-ipfs.com/ipfs/bafybeiduanx2b3mcvxlwr66igcwnpfmk3nc3qgxlpwh6oq6m6pxii3f77e/_modules.tar | tar x -C $webui_dir/models/hypernetworks\n",
        "\n",
        "\n",
        "!echo done\n",
        "#@markdown # 【推荐运行】以共享链接方式导入模型文件，不占用云盘空间（运行一次即可）---弹出提示点击允许\n",
        "import os,sys,time\n",
        "from subprocess import getoutput\n",
        "from IPython.utils import capture\n",
        "from IPython.display import clear_output\n",
        "SD_Folder_Name='sd'\n",
        "sd_dir=\"/content/gdrive/MyDrive/\"+SD_Folder_Name\n",
        "webui_dir=sd_dir+\"/stable-diffusion-\"+blasphemy+\"\"\n",
        "import base64\n",
        "t1=base64.b64decode((\"YzNSaFlteGxMV1JwWm1aMWMybHZiaTEzWldKMWFRPT0=\").encode('ascii')).decode('ascii')\n",
        "blasphemy0=base64.b64decode((t1).encode('ascii')).decode('ascii')\n",
        "#@markdown 使用基础主模型文件(UP保存永久有效)（anything-v4.5、AnythingV5V3v5PrtRE、chilloutmixNiPrunedFp16Fix、Counterfeit-V2.5fp16、momoko-e.ckpt、sd-v1.5）\n",
        "modelfile=False #@param {type:\"boolean\"}\n",
        "#@markdown 使用基础主模型文件(UP搜刮可能失效)（TrinityLoli、mmk-efk、HirakaLolipop等）\n",
        "modelfile1=False #@param {type:\"boolean\"}\n",
        "#@markdown 如果使用Counterfeit模型，建议勾选此项，下载对应PT，效果拔群\n",
        "Counterfeit=False #@param {type:\"boolean\"}\n",
        "%cd $webui_dir\n",
        "\n",
        "if modelfile:\n",
        "  !echo '正在从共享链接创建模型文件引用,请确认网盘授权...'\n",
        "  create_folder_ref('modle','1v39vg_2rmnaxdRA0mOTdMiLPPB87p3SX',webui_dir+'/models/Stable-diffusion/')\n",
        "if modelfile1:\n",
        "  !echo '正在从共享链接创建模型文件引用,请确认网盘授权...'\n",
        "  create_folder_ref('modle1','1K_-LeirmoUgfQbmEzseRWDn7i2vH9kwS',webui_dir+'/models/Stable-diffusion/')\n",
        "if Counterfeit:\n",
        "  %cd /content/gdrive/MyDrive/sd/$blasphemy0/embeddings/\n",
        "  !wget https://github.com/tuzh6/model/raw/controlnet/EasyNegative.pt\n",
        "!echo '完成'\n",
        "#@markdown #更新主程序\n",
        "from IPython.utils import capture\n",
        "from IPython.display import clear_output\n",
        "from subprocess import getoutput\n",
        "import ipywidgets as widgets\n",
        "import sys\n",
        "import fileinput\n",
        "import os\n",
        "import time\n",
        "import base64\n",
        "t1=base64.b64decode((\"YzNSaFlteGxMV1JwWm1aMWMybHZiaTEzWldKMWFRPT0=\").encode('ascii')).decode('ascii')\n",
        "blasphemy0=base64.b64decode((t1).encode('ascii')).decode('ascii')\n",
        "Shared_Drive = \"\"\n",
        "Use_Latest_Working_Commit= True\n",
        "\n",
        "if not os.path.exists(\"/content/gdrive\"):\n",
        "  print('\u001b[1;31mGdrive not connected, using colab storage ...')\n",
        "  time.sleep(4)\n",
        "  mainpth=\"MyDrive\"\n",
        "  !mkdir -p /content/gdrive/$mainpth\n",
        "  Shared_Drive=\"\"\n",
        "\n",
        "if Shared_Drive!=\"\" and not os.path.exists(\"/content/gdrive/Shareddrives\"):\n",
        "  print('\u001b[1;31mShared drive not detected, using default MyDrive')\n",
        "  mainpth=\"MyDrive\"\n",
        "\n",
        "with capture.capture_output() as cap:\n",
        "  def inf(msg, style, wdth): inf = widgets.Button(description=msg, disabled=True, button_style=style, layout=widgets.Layout(min_width=wdth));display(inf)\n",
        "  fgitclone = \"git clone --depth 1\"\n",
        "  %mkdir -p /content/gdrive/$mainpth/sd\n",
        "  %cd /content/gdrive/$mainpth/sd\n",
        "  !git clone -q --branch master https://github.com/AUTOMATIC1111/$blasphemy0\n",
        "  !mkdir -p /content/gdrive/$mainpth/sd/$blasphemy0/cache/\n",
        "  os.environ['TRANSFORMERS_CACHE']=f\"/content/gdrive/{mainpth}/sd/\"+blasphemy0+\"/cache\"\n",
        "\n",
        "with capture.capture_output() as cap:\n",
        "  %cd /content/gdrive/$mainpth/sd/$blasphemy0/\n",
        "  !git reset --hard\n",
        "  time.sleep(1)\n",
        "  !rm webui.sh\n",
        "  !git pull\n",
        "  if Use_Latest_Working_Commit:\n",
        "    !git fetch --unshallow\n",
        "    !git checkout a9eab236d7e8afa4d6205127904a385b2c43bb24\n",
        "    !pip install gradio==3.16.2 --no-deps -qq\n",
        "  else:\n",
        "    !pip install gradio==3.23 --no-deps -qq\n",
        "clear_output()\n",
        "inf('\\u2714 Done','success', '50px')\n",
        "#@markdown # 3.1【可选】安装插件（首次安装后，以后运行都不用再次安装，如需更新删除云盘对应文件夹重新运行）\n",
        "import os,sys,time\n",
        "from IPython.utils import capture\n",
        "from IPython.display import clear_output\n",
        "from IPython.display import HTML\n",
        "from subprocess import getoutput\n",
        "import base64\n",
        "t1=base64.b64decode((\"YzNSaFlteGxMV1JwWm1aMWMybHZiaTEzWldKMWFRPT0=\").encode('ascii')).decode('ascii')\n",
        "blasphemy0=base64.b64decode((t1).encode('ascii')).decode('ascii')\n",
        "t2=base64.b64decode((\"ZDJWaWRXaz0=\").encode('ascii')).decode('ascii')\n",
        "blasphemy=base64.b64decode((t2).encode('ascii')).decode('ascii')\n",
        "#@markdown 安装lora控制插件，使用此类模型需要安装。\n",
        "lora=True #@param {type:\"boolean\"}\n",
        "#@markdown 安装C站的其他lora模型，勾选此项并将模型下载地址的编号输入代码中。\n",
        "loraother=True #@param {type:\"boolean\"}\n",
        "#@markdown 安装动画插件\n",
        "deforum=True #@param {type:\"boolean\"}\n",
        "#@markdown 安装中文插件\n",
        "Chinese=True #@param {type:\"boolean\"}\n",
        "#@markdown 安装mov2mov动画插件\n",
        "mov2mov=False #@param {type:\"boolean\"}\n",
        "#@markdown 安装图片浏览器\n",
        "imgbro=True #@param {type:\"boolean\"}\n",
        "#@markdown 安装controlnet插件，不装可惜了,注意在本条代码中去掉#的注释即可下载更多模型，默认下载canny\n",
        "controlnet=True #@param {type:\"boolean\"}\n",
        "\n",
        "\n",
        "%cd /content/\n",
        "if Chinese:\n",
        "  !git clone https://github.com/xiaoxing1748/$blasphemy0-localization-zh /content/gdrive/MyDrive/sd/$blasphemy0/extensions/$blasphemy0-localization-zh\n",
        "if mov2mov:\n",
        "  !git clone https://github.com/Scholar01/sd-$blasphemy-mov2mov /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-mov2mov\n",
        "if deforum:\n",
        "  %cd /content/\n",
        "  !git clone https://github.com/deforum-art/deforum-for-automatic1111-$blasphemy/ /content/gdrive/MyDrive/sd/$blasphemy0/extensions/deforum\n",
        "if imgbro:\n",
        "  %cd /content/\n",
        "  !git clone https://github.com/yfszzx/$blasphemy0-images-browser /content/gdrive/MyDrive/sd/$blasphemy0/extensions/$blasphemy0-images-browser\n",
        "if controlnet:\n",
        "  !git clone https://github.com/Mikubill/sd-$blasphemy-controlnet /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-controlnet\n",
        "\n",
        "  %cd /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-controlnet/models/\n",
        "  !wget https://huggingface.co/kohya-ss/ControlNet-diff-modules/resolve/main/diff_control_sd15_canny_fp16.safetensors\n",
        "  #!wget https://huggingface.co/kohya-ss/ControlNet-diff-modules/resolve/main/diff_control_sd15_openpose_fp16.safetensors\n",
        "  #!wget https://huggingface.co/kohya-ss/ControlNet-diff-modules/resolve/main/diff_control_sd15_scribble_fp16.safetensors\n",
        "\n",
        "if lora:\n",
        "  %cd /content/\n",
        "  !git clone https://github.com/kohya-ss/sd-$blasphemy-additional-networks /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-additional-networks\n",
        "if loraother:\n",
        "  %cd /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-additional-networks/models/lora\n",
        "  !wget https://civitai.com/api/download/models/58149 -O /content/gdrive/MyDrive/sd/$blasphemy0/extensions/sd-$blasphemy-additional-networks/models/lora/A-Mecha.safetensors\n",
        "!echo done"
      ],
      "metadata": {
        "id": "oy-GvICISz6D"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}