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

图像生成提示词参考

https://arthub.ai/

作品欣赏

SDK代码下载地址：

Github链接
Gitee链接

gpu提示词diffusioncpustablediffusionjava文生图图像生成pytorchpromptgit图生图githubsdk测试数据url