如何使用NovelAI畫馬圖(教程)

前幾天看EqD上面介紹了下這個模型(https://huggingface.co/AstraliteHeart/pony-diffusion),看上去還不錯,就下載了看看
先上幾張生成出來的馬圖(注釋里是prompt)



這個模型是拿derpibooru上500+分的safe和suggestive圖,根據(jù)他們的tags來訓(xùn)練的。
理論上圖最多的小馬生成的質(zhì)量最好,所以g5小馬和背景小馬還是別想咯(嗚嗚我的pipp

首先安裝Python
https://www.python.org/downloads/
無腦裝最新版本就完事了
如果你有現(xiàn)成的python請確保是3.7以上的版本

然后去這個網(wǎng)站下載Pytorch
https://pytorch.org/get-started/locally/
下滑來到這個頁面

Pytorch build選Stable
Your OS選你的系統(tǒng)
Package選Pipp(如果你沒有裝其他的package manager的話)
Language選Python
Compute Platform根據(jù)你的顯卡選:
如果你是Geforce 8系列及以后的顯卡,選cuda 11.6。
如果你使用Linux系統(tǒng)并且使用RX系列及以后的顯卡,選ROCm 5.1.1
其他情況只能用cpu渲染
選完后復(fù)制Run this command里面的內(nèi)容到你的終端/命令提示符里執(zhí)行,等待執(zhí)行完成。(需要下載1GiB+的文件)
然后繼續(xù)裝其他dependency
pip3 install transformer diffusers?
(建議sudo ln /usr/bin/pip /usr/bin/pipp以獲得最佳體驗(逃)

接下來在桌面創(chuàng)建一個名叫test.py的文件(其實哪里都行,名字也可以隨便?。?,內(nèi)容如下
(Adapted from https://huggingface.co/AstraliteHeart/pony-diffusion)
根據(jù)注釋來修改代碼,默認是使用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(工作時不適宜)內(nèi)容被和諧了,換個prompt試試,或者再生成一次
?
Q:CUDA out of memory. Tried to allocate 512.00 MiB……
A:爆顯存了。你需要6GiB以上的顯存,如果沒有,請使用CPU渲染
?
如果有其他的疑惑或者問題可以在評論區(qū)里補充,我會第一時間補上。
本人剛開始接觸AI繪畫領(lǐng)域,如果有紕漏還請大佬們指出糾正(