鸣谢
参照这位博主的笔记成功部署了ai绘画工具,在此表示感谢!
准备工作
(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
运行代码
import os
os.kill(os.getpid(), 9) # 必要,报错很正常
(2)下载Stable-Diffusion-Webui
!mv stable-diffusion-webui-cn stable–diffusion–webui
%cd stable–diffusion–webui
!git pull
(3)下载加载模型资源
#@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(“完成!”)
这里是从我们谷歌云盘中加载模型,注意我们文件夹和文件的命名和代码中是否一致,稍微不一样可能就找不到文件。
drive.mount(/content/drive)
加载这些资源可能会使图片质量更高。
#@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)安装依赖
%cd /content/stable-diffusion-webui
!COMMANDLINE_ARGS=”–exit” REQS_FILE=”requirements.txt” python launch.py
更换模型配置文件
#@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)运行
!COMMANDLINE_ARGS=”–share –gradio-debug” REQS_FILE=”requirements.txt” python launch.py
成功后会出现 Running on public URL: xxx ,访问即可
如果想把生成的图片放到谷歌云盘: