diff --git a/bootstrap.sh b/bootstrap.sh index 4a9842c..a57ae05 100755 --- a/bootstrap.sh +++ b/bootstrap.sh @@ -1,5 +1,9 @@ #!/bin/bash git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui /shared/stable-diffusion-webui +cd /shared +ln -s ./stable-diffusion-webui/models models +ln -s ./stable-diffusion-webui/outputs outputs +ln -s ./stable-diffusion-webui/embeddings embeddings cd /shared/stable-diffusion-webui exec ./webui.sh \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 735720d..f8e3dcb 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -3,11 +3,15 @@ services: stable-diffusion: build: . environment: + # for 5700xt compat + # ref: https://old.reddit.com/r/StableDiffusion/comments/ww436j/howto_stable_diffusion_on_an_amd_gpu/imcpmo8/ - HSA_OVERRIDE_GFX_VERSION=10.3.0 - - COMMANDLINE_ARGS=--precision full --no-half - network_mode: host - # for 5700xt compat - # ref: https://old.reddit.com/r/StableDiffusion/comments/ww436j/howto_stable_diffusion_on_an_amd_gpu/imcpmo8/ + - PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512 + - COMMANDLINE_ARGS=--listen --theme=dark --precision full --no-half + # - COMMANDLINE_ARGS=--listen --theme=dark --precision full --no-half --medvram + # - COMMANDLINE_ARGS=--listen --theme=dark --use-cpu all --precision full --no-half --skip-torch-cuda-test + ports: + - 7860:7860 devices: - /dev/kfd - /dev/dri