zepheris/stable-diffusion-webui-docker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change capabilities in docker-compose (#497)

Browse Source 
#479
This commit is contained in:
AbdBarho 2023-06-02 21:29:00 +02:00 committed by GitHub
parent 6695c23150
commit 660c098da0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yml
View File

@ -13,7 +13,7 @@ x-base_service: &base_service
devices: devices:
- driver: nvidia - driver: nvidia
device_ids: ['0'] device_ids: ['0']
capabilities: [gpu] capabilities: [compute, utility]
name: webui-docker name: webui-docker

1
services/AUTOMATIC1111/Dockerfile
View File

@ -88,7 +88,6 @@ RUN \
git config --global --add safe.directory '*' git config --global --add safe.directory '*'
WORKDIR ${ROOT} WORKDIR ${ROOT}
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_VISIBLE_DEVICES=all
ENV CLI_ARGS="" ENV CLI_ARGS=""
EXPOSE 7860 EXPOSE 7860

2
services/comfy/Dockerfile
View File

@ -38,7 +38,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
COPY . /docker/ COPY . /docker/
RUN cp /docker/extra_model_paths.yaml ${ROOT} RUN cp /docker/extra_model_paths.yaml ${ROOT}
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_VISIBLE_DEVICES=all
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS="" ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
EXPOSE 7860 EXPOSE 7860
ENTRYPOINT ["/docker/entrypoint.sh"] ENTRYPOINT ["/docker/entrypoint.sh"]

2
services/invoke/Dockerfile
View File

@ -42,7 +42,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
COPY . /docker/ COPY . /docker/
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_VISIBLE_DEVICES=all
ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR=/data/config/invoke CLI_ARGS="" ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR=/data/config/invoke CLI_ARGS=""
EXPOSE 7860 EXPOSE 7860