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

NovelAI指南2年前 (2023)发布 一起用AI
796 0 0

鸣谢

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

准备工作

(1)下载训练好的模型

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

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

(2)准备好一个谷歌账号

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

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

新建文件

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

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

修改访问限制

部署步骤

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

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

(1)首先打开colab的gpu支持

Runtime-Change runtime type-GPU

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

运行代码

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

(2)下载Stable-Diffusion-Webui

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

(3)下载加载模型资源

#@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/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(“完成!”)

这里是从我们谷歌云盘中加载模型,注意我们文件夹和文件的命名和代码中是否一致,稍微不一样可能就找不到文件。

使用Colab云端部署AI绘画工具
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/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

(5)运行

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

成功后会出现 Running on public URL: xxx ,访问即可

如果想把生成的图片放到谷歌云盘:

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

检验成果!

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

© 版权声明

相关文章