mirror of
https://github.com/open-webui/open-webui
synced 2024-11-16 21:42:58 +00:00
32 lines
1.1 KiB
YAML
32 lines
1.1 KiB
YAML
|
# This is an overlay that spins up stable-diffusion-webui for integration testing
|
||
|
# This is not designed to be used in production
|
||
|
services:
|
||
|
stable-diffusion-webui:
|
||
|
# Not built for ARM64
|
||
|
platform: linux/amd64
|
||
|
image: ghcr.io/neggles/sd-webui-docker:latest
|
||
|
restart: unless-stopped
|
||
|
environment:
|
||
|
CLI_ARGS: "--api --use-cpu all --precision full --no-half --skip-torch-cuda-test --ckpt /empty.pt --do-not-download-clip --disable-nan-check --disable-opt-split-attention"
|
||
|
PYTHONUNBUFFERED: "1"
|
||
|
TERM: "vt100"
|
||
|
SD_WEBUI_VARIANT: "default"
|
||
|
# Hack to get container working on Apple Silicon
|
||
|
# Rosetta creates a conflict ${HOME}/.cache folder
|
||
|
entrypoint: /bin/bash
|
||
|
command:
|
||
|
- -c
|
||
|
- |
|
||
|
export HOME=/root-home
|
||
|
rm -rf $${HOME}/.cache
|
||
|
/docker/entrypoint.sh python -u webui.py --listen --port $${WEBUI_PORT} --skip-version-check $${CLI_ARGS}
|
||
|
volumes:
|
||
|
- ./test/test_files/image_gen/sd-empty.pt:/empty.pt
|
||
|
|
||
|
open-webui:
|
||
|
environment:
|
||
|
ENABLE_IMAGE_GENERATION: "true"
|
||
|
AUTOMATIC1111_BASE_URL: http://stable-diffusion-webui:7860
|
||
|
IMAGE_SIZE: "64x64"
|
||
|
IMAGE_STEPS: "3"
|