使用飞桨快速部署stable diffusion模型

skywalk81632024-04-25 1:38

这可以说是最快的部署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