itsnachocheese
27 lines
682 B
YAML
27 lines
682 B
YAML
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=--api --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"
|