当前位置:   article > 正文

人工智能AI系列 - java 版的stable diffusion 图像生成_java集成stable diffusion

java集成stable diffusion

图像生成

  • 文生图:输入提示词(仅支持英文),生成图片(仅支持英文)
  • GPU版本 StableDiffusionGPU.java
  • CPU版本 StableDiffusionCPU.java
  • 图生图:根据图片及提示词(仅支持英文)生成图片
  • GPU版本 Img2ImgStableDiffusionGPU.java
  • 显卡CUDA:11.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:
  • 生成图片效果:

图像生成提示词参考

作品欣赏

SDK代码下载地址:

Github链接
Gitee链接

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/552691
推荐阅读
相关标签
  

闽ICP备14008679号