人工智能AI系列 - 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：
• 生成图片效果：

图像生成提示词参考

作品欣赏

**开源代码下载地址：

**