这可以说是最快的部署stable diffusion模型的方法了,而且星河社区还有免费的GPU算力提供,这样再也不用担心没有算力了!
操作方法:
# 安装相关库
pip install docarray==0.21.0
pip install paddlehub
# 初始化
from PIL import Image
import paddlehub as hub
module = hub.Module(name='stable_diffusion')
# 开始画图,先画个灯塔
result = module.generate_image(text_prompts="A beautiful painting of a singular lighthouse, shining its light across a tumultuous sea of blood by greg rutkowski and thomas kinkade, Trending on artstation.", output_dir='stable_diffusion_out')
# 用中文试试,发现效果不好,还是改成了英文
result = module.generate_image(text_prompts="Blood elf mage of world of warcraft, using fireball spell", output_dir='stable_diffusion_out')
可以看到画的图还是不错的
在V100下22秒输出一张图。
注意,模型限制77个字符的输入,超出会报错:is too long for context length 77