AIAS/9_aigc/stable_diffusion/README_cn.md
2023-08-20 23:24:44 +08:00

1.7 KiB
Executable File
Raw Blame History

官网:

官网链接

图像生成

  • 文生图:输入提示词(仅支持英文),生成图片(仅支持英文)

  • GPU版本 StableDiffusionGPU.java

  • CPU版本 StableDiffusionCPU.java

  • 图生图:根据图片及提示词(仅支持英文)生成图片

  • GPU版本 Img2ImgStableDiffusionGPU.java

  • 显卡CUDA11.7版本

  • 参考测试数据:分辨率 512*512 25步 CPU(i5处理器) 5分钟。 3060显卡20秒。

模型下载

  • cpu模型保存位置stable_diffusion/pytorch_cpu
  • gpu模型保存位置stable_diffusion/pytorch_gpu

文生图测试

  • 提示词 prompt a photo of an astronaut riding a horse on mars
  • 反向提示词 negative_prompt
  • 生成图片效果:

图像生成提示词参考

作品欣赏

Git地址

Github链接
Gitee链接

帮助文档: