cjwbw/stable-diffusion-v2

Public sd-v2 with diffusers, test version!
219.6K runs
Demo API Examples Versions (e5e1fd33)

Examples

View more examples

Run time and cost

Predictions run on Nvidia A100 (40GB) GPU hardware. Predictions typically complete within 4 seconds.

weights from: https://huggingface.co/stabilityai/stable-diffusion-2, fp32 version code for the demo: https://github.com/chenxwh/cog-stable-diffusion/tree/sd-v2

TEST version!

This stable-diffusion-2 model is resumed from stable-diffusion-2-base (512-base-ema.ckpt) and trained for 150k steps using a v-objective on the same dataset. Resumed for another 140k steps on 768x768 images.

Compared to the mainline stable-diffusion https://replicate.com/stability-ai/stable-diffusion, the differences are: - seems only DDIM, K_EULER, and DPMSolverMultistep schedulers work properly - removed image inpainting pipeline - Default size set 768X768 - less num_output is allowed to fit to on V100