From 792db4d1c3e74ae71cff379077222d1d9bae2a80 Mon Sep 17 00:00:00 2001 From: Mrau Hu Date: Sat, 12 Nov 2022 19:31:16 +0300 Subject: [PATCH] Created `TORCH_COMMAND` argument, with same name as in Stable Diffusion Web UI docs --- services/AUTOMATIC1111/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile index c62f19d..09d1b84 100644 --- a/services/AUTOMATIC1111/Dockerfile +++ b/services/AUTOMATIC1111/Dockerfile @@ -34,8 +34,8 @@ SHELL ["/bin/bash", "-ceuxo", "pipefail"] ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1 -RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 - +ARG TORCH_COMMAND="pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113" +RUN ${TORCH_COMMAND} RUN apt-get update && apt install fonts-dejavu-core rsync git jq moreutils -y && apt-get clean