stable-diffusion-rocm
stable-diffusion-rocm copied to clipboard
Stable Diffusion Dockerfile for ROCm
Run Stable Diffusion on an AMD card, using this method. Tested on my RX 6900 XT.
- Obtain
sd-v1-4.ckpt
and put it inmodels/
. - Run
./build-rocm
to build the Docker image. - Run
./run-rocm
to run a shell in the Docker container. - Inside the container, you can do e.g.
python scripts/txt2img.py --outdir /output --plms --seed $RANDOM$RANDOM --prompt "a unicorn riding a purple tricycle"
.