fboulnois / stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.
GNU Affero General Public License v3.0
735 stars 127 forks source link

Add stable unCLIP variations and VAE slicing #79

Closed fboulnois closed 1 year ago

fboulnois commented 1 year ago

Adds stable unCLIP variations which creates different versions of an existing image with a text prompt:

./build.sh run --model 'stabilityai/stable-diffusion-2-1-unclip-small' \
  --image parakeet_eating_spaghetti.png \
  --prompt 'An impressionist painting of a parakeet eating spaghetti in the desert'

Also adds VAE slicing to lower memory use when generating batches of images:

./build.sh run --samples 2 --iters 2 \
  --skip --vae-slicing --xformers-memory-efficient-attention \
  --prompt "abstract art"

Other changes: