diff --git a/.github/workflows/webui.v2.6.yml b/.github/workflows/webui.v2.6.yml index 5504299..263e4e8 100644 --- a/.github/workflows/webui.v2.6.yml +++ b/.github/workflows/webui.v2.6.yml @@ -29,8 +29,8 @@ jobs: uses: docker/build-push-action@v5 with: context: . - file: ./Dockerfile.v2.6.gpu + file: ./Dockerfile.test no-cache: true push: true - tags: camenduru/webui:v2.6.gpu + tags: camenduru/webui:test diff --git a/Dockerfile.test b/Dockerfile.test new file mode 100644 index 0000000..eceddb7 --- /dev/null +++ b/Dockerfile.test @@ -0,0 +1,25 @@ +FROM nvidia/cuda:12.2.0-base-ubuntu22.04 +ENV DEBIAN_FRONTEND noninteractive + +RUN apt-get update -y && \ + apt-get install -y aria2 libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && \ + pip install -q torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 torchtext==0.15.2 torchdata==0.6.1 --extra-index-url https://download.pytorch.org/whl/cu118 && \ + pip install xformers==0.0.20 triton==2.0.0 packaging==23.1 gdown && \ + adduser --disabled-password --gecos '' user && \ + mkdir /content && \ + chown -R user:user /content + +WORKDIR /content +USER user + +RUN git clone -b v2.6 https://github.com/camenduru/stable-diffusion-webui /content/test && \ + git clone https://github.com/etherealxx/batchlinks-webui /content/test/extensions/batchlinks-webui && \ + git clone https://github.com/Mikubill/sd-webui-controlnet /content/test/extensions/sd-webui-controlnet && \ + git clone https://github.com/zanllp/sd-webui-infinite-image-browsing /content/test/extensions/sd-webui-infinite-image-browsing && \ + git clone https://github.com/BlafKing/sd-civitai-browser-plus /content/test/extensions/sd-civitai-browser-plus && \ + sed -i -e 's/ start()/ #start()/g' /content/test/launch.py && \ + cd /content/test && \ + python launch.py --skip-torch-cuda-test && \ + git reset --hard + +CMD cd /content/test && python launch.py --xformers --cors-allow-origins=* --api