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

新闻 2023-02-21

437 0

鸣谢

参照这位博主的笔记成功部署了ai绘画工具，在此表示感谢！

绘画AI NovelAI部署教程教你如何在本地搭建 NovelAImod.3dmgame.com/read/24

准备工作

（1）下载训练好的模型

这里是一个简化版的小模型，免费版的colab只能带的动小模型：https://drive.google.com/file/d/1NaJcM3ba7gUG4cDUtHJSLalcDsAUVLRr/view

完整版大模型的下载地址：https://drive.google.com/file/d/1e5Wu3ahBvYbhl4DfQN6h7q_UDZdC6y3t/view

（2）准备好一个谷歌账号

在自己的谷歌云盘中新建一个folder，命名为NovelAI，将下载的模型导入进入，这个过程可能会持续几十分钟。

新建文件

注意需要修改文件的访问限制。

修改访问限制

部署步骤

以下部署步骤是根据一开始引用的文章中的模板来的，模板中的代码自己一步一步运行就好，但是可能存在命名不一致导致的错误，所以这里简单做个讲解。

模板地址：https://colab.research.google.com/drive/1GRIQehJ__ShyGZUeBpXxJNbecA-OHx1u?usp=sharing

（1）首先打开colab的gpu支持

Runtime-Change runtime type-GPU

运行代码

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

（2）下载Stable-Diffusion-Webui

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

（3）下载加载模型资源

#@title 加载模型 (小模型) 模型链接又炸了
#@markdown **请自行下载小模型，并上传到google云盘** 将模型放置到云盘的 `NovelAI` 目录下，并改名为 `animefull-final-pruned.ckpt`。免费版的colab只能加载得动小模型，效果也不错
#@markdown 【磁链】magnet:?xt=urn:btih:5bde442da86265b670a3e5ea3163afad2c6f8ecc
#@markdown [【小模型 GoogleDrive】](https://drive.google.com/file/d/1NaJcM3ba7gUG4cDUtHJSLalcDsAUVLRr/view?usp=sharing)
#@markdown [【小模型 OneDrive】](https://1drv.ms/u/s!Aj0eXAYia5WNjdIHWbXojC5sAJp3Ow?e=BXR6Ne)
#@markdown [【大模型 GoogleDrive】](https://drive.google.com/file/d/1e5Wu3ahBvYbhl4DfQN6h7q_UDZdC6y3t/view?usp=sharing)
%cd /content/drive/MyDrive

from google.colab import files
import os

if 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 [【hypernetworks GoogleDrive】](https://drive.google.com/file/d/1mYRDVMJdyd1iH1cooUQeKUUAAdXA61Hn/view?usp=sharing)
#@markdown [【hypernetworks OneDrive】](https://1drv.ms/u/s!Aj0eXAYia5WNjdIIcSL-BiUAQMNm6w?e=Guebb9)
%cd /content/drive/MyDrive

from google.colab import files
import os

if 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("完成")

（4）安装依赖

#@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