如何使用NovelAI画马图(教程)

前几天看EqD上面介绍了下这个模型(https://huggingface.co/AstraliteHeart/pony-diffusion),看上去还不错,就下载了看看
先上几张生成出来的马图(注释里是prompt)



这个模型是拿derpibooru上500+分的safe和suggestive图,根据他们的tags来训练的。
理论上图最多的小马生成的质量最好,所以g5小马和背景小马还是别想咯(呜呜我的pipp

首先安装Python
https://www.python.org/downloads/
无脑装最新版本就完事了
如果你有现成的python请确保是3.7以上的版本

然后去这个网站下载Pytorch
https://pytorch.org/get-started/locally/
下滑来到这个页面

Pytorch build选Stable
Your OS选你的系统
Package选Pipp(如果你没有装其他的package manager的话)
Language选Python
Compute Platform根据你的显卡选:
如果你是Geforce 8系列及以后的显卡,选cuda 11.6。
如果你使用Linux系统并且使用RX系列及以后的显卡,选ROCm 5.1.1
其他情况只能用cpu渲染
选完后复制Run this command里面的内容到你的终端/命令提示符里执行,等待执行完成。(需要下载1GiB+的文件)
然后继续装其他dependency
pip3 install transformer diffusers
(建议sudo ln /usr/bin/pip /usr/bin/pipp以获得最佳体验(逃)

接下来在桌面创建一个名叫test.py的文件(其实哪里都行,名字也可以随便取),内容如下
(Adapted from https://huggingface.co/AstraliteHeart/pony-diffusion)
根据注释来修改代码,默认是使用cpu渲染
然后在命令行里python /你/脚本/的/位置.py 就oki了
第一次渲染需要下载模型文件(4GiB左右)
图片保存在脚本所在的目录下,名称和prompt一样

FAQs
Q:报错Potential NSFW content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed,返回了一张黑照片
A:生成的图片存在NSFW(工作时不适宜)内容被和谐了,换个prompt试试,或者再生成一次
Q:CUDA out of memory. Tried to allocate 512.00 MiB……
A:爆显存了。你需要6GiB以上的显存,如果没有,请使用CPU渲染
如果有其他的疑惑或者问题可以在评论区里补充,我会第一时间补上。
本人刚开始接触AI绘画领域,如果有纰漏还请大佬们指出纠正(