/stable-diffusion-rocm

A fork of stable-diffusion-rocm that works for 6700/6700xt cards.

Primary LanguageDockerfile

Stable Diffusion Dockerfile for ROCm

Run Stable Diffusion on an AMD card, using this method. Tested on my RX 6700 (non XT). Uses basujindal's stable diffusion fork for lower memory usage.

  1. Obtain sd-v1-4.ckpt and put it in rocm/
  2. Run ./build-rocm to build the Docker image
  3. Run ./run-rocm to run a shell in the Docker container
  4. Inside the container, you can run commands to generate images like: python optimizedSD/optimized_txt2img.py --outdir /output --prompt "a unicorn riding a purple tricycle"