created ROCm-compatible docker image for automatic1111's stablediffusion webui
This commit is contained in:
@@ -0,0 +1,26 @@
|
||||
services:
|
||||
stable-diffusion-webui:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
memory: 16G
|
||||
cpus: 16
|
||||
container_name: stable-diffusion-webui
|
||||
restart: unless-stopped
|
||||
environment:
|
||||
- HSA_OVERRIDE_GFX_VERSION=11.0.0
|
||||
- HIP_VISIBLE_DEVICES=0
|
||||
- PYTORCH_ROCM_ARCH=gfx1100
|
||||
- CLI_ARGS=--listen --port 7860 --opt-sdp-attention --no-half-vae --skip-torch-cuda-test
|
||||
ports:
|
||||
- "7860:7860"
|
||||
volumes:
|
||||
- ./models:/app/models
|
||||
- ./outputs:/app/outputs
|
||||
- ./extensions:/app/extensions
|
||||
devices:
|
||||
- "/dev/kfd:/dev/kfd"
|
||||
- "/dev/dri:/dev/dri"
|
||||
Reference in New Issue
Block a user