当前位置:AIGC资讯 > AIGC > 正文

人工智能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链接

更新时间 2023-12-04