使用Colab云端部署AI绘画工具

#@markdown 会导致colab崩溃，不用管 （必要）
import os
os.kill(os.getpid(), 9) # 必要，报错很正常

!git clone https://github.com/3DMXM/stable-diffusion-webui-cn
!mv stable-diffusion-webui-cn stable-diffusion-webui
%cd stable-diffusion-webui
!git pull

#@title 加载模型 (小模型) 模型链接又炸了
#@markdown **请自行下载小模型，并上传到google云盘** 将模型放置到云盘的 `NovelAI` 目录下，并改名为 `animefull-final-pruned.ckpt`。免费版的colab只能加载得动小模型，效果也不错
#@markdown <br> 【磁链】magnet:?xt=urn:btih:5bde442da86265b670a3e5ea3163afad2c6f8ecc
#@markdown <br> [【小模型 GoogleDrive】](https://drive.google.com/file/d/1NaJcM3ba7gUG4cDUtHJSLalcDsAUVLRr/view?usp=sharing)
#@markdown <br> [【小模型 OneDrive】](https://1drv.ms/u/s!Aj0eXAYia5WNjdIHWbXojC5sAJp3Ow?e=BXR6Ne)
#@markdown <br> [【大模型 GoogleDrive】](https://drive.google.com/file/d/1e5Wu3ahBvYbhl4DfQN6h7q_UDZdC6y3t/view?usp=sharing)
%cd /content/drive/MyDrivefrom google.colab import files
import osif not os.path.exists(NovelAI/animefull-final-pruned.ckpt):
!mkdir NovelAi
%cd NovelAi
print("开始下载模型...")
!gdown "1NaJcM3ba7gUG4cDUtHJSLalcDsAUVLRr" # google云盘的文件下载次数过多被限制了
# !wget https://plastic-global-sundial.glitch.me/animefull-final-pruned.ckpt # onedrive文件也寄了

%cd /content/drive/MyDrive
if not os.path.exists(NovelAI/animefull-final-pruned.ckpt):
print("未找到模型")
!cp /content/drive/MyDrive/NovelAI/animefull-final-pruned.ckpt /content/stable-diffusion-webui/models/Stable-diffusion/
!cp /content/drive/MyDrive/NovelAI/sd-v1-5-inpainting.ckpt /content/stable-diffusion-webui/models/Stable-diffusion/
print("完成!")

from google.colab import drive
drive.mount(/content/drive)

#@title 加载资源 (可选)
#@markdown 可能会使图片质量更高。自行下载的话，把压缩包也放到云盘的 `NovelAI` 目录下
#@markdown <br> [【hypernetworks GoogleDrive】](https://drive.google.com/file/d/1mYRDVMJdyd1iH1cooUQeKUUAAdXA61Hn/view?usp=sharing)
#@markdown <br> [【hypernetworks OneDrive】](https://1drv.ms/u/s!Aj0eXAYia5WNjdIIcSL-BiUAQMNm6w?e=Guebb9)
%cd /content/drive/MyDrivefrom google.colab import files
import osif not os.path.exists(NovelAI/hypernetworks.zip):
!mkdir NovelAi
%cd NovelAi
print("开始下载资源...")
!gdown "1mYRDVMJdyd1iH1cooUQeKUUAAdXA61Hn"
# !wget https://plastic-global-sundial.glitch.me/hypernetworks.zip

!cp /content/drive/MyDrive/NovelAi/hypernetworks.zip /content/stable-diffusion-webui/models/
%cd /content/stable-diffusion-webui/models
!unzip hypernetworks.zip -d hypernetworks
print("完成")

#@title 安装依赖
%cd /content/stable-diffusion-webui
!COMMANDLINE_ARGS="--exit" REQS_FILE="requirements.txt" python launch.py

#@title 更换模型配置文件
#@markdown 使用的是小模型的配置文件
%cd /content/stable-diffusion-webui/repositories/stable-diffusion/configs/stable-diffusion
!mv v1-inference.yaml v1-inference-or.yaml
!gdown "1HwkqvjMiP4bjcPFcaPL9thxaV9G3JV_J" --output v1-inference.yaml

%cd /content/stable-diffusion-webui
!COMMANDLINE_ARGS="--share --gradio-debug" REQS_FILE="requirements.txt" python launch.py

!cp -r /content/stable-diffusion-webui/outputs /content/drive/MyDrive/NovelAi/