图像生成
- 文生图：输入提示词（仅支持英文），生成图片（仅支持英文）
- 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代码下载地址：
文章出处登录后可见！
已经登录？立即刷新
