图像生成
文生图:输入提示词(仅支持英文),生成图片(仅支持英文) 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链接