Update Dockerfile

Dockerfile

@@ -1,49 +1,50 @@
+
+# Dockerfile Public Nightly A10G
+
+# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.7.1/ubuntu2204/devel/cudnn8/Dockerfile
 FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+ENV DEBIAN_FRONTEND noninteractive
 
-ENV DEBIAN_FRONTEND=noninteractive \
-    TZ=America/Los_Angeles
-
-ARG USE_PERSISTENT_DATA
-
-RUN apt-get update && apt-get install -y \
-    git libgl1 libglib2.0-0 \
-    make build-essential libssl-dev zlib1g-dev \
-    libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
-    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git git-lfs  \
-    ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
-    && rm -rf /var/lib/apt/lists/* \
-    && git lfs install \
-    apt-get install nvidia-container-runtime
-    
 WORKDIR /content
 
-# User
-RUN useradd -m -u 1000 user
+RUN apt-get update -y && apt-get upgrade -y && apt-get install -y libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && pip3 install --upgrade pip
+RUN pip install xformers==0.0.27
+RUN pip install --pre triton
+RUN pip install numexpr
+
+RUN git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
+RUN sed -i -e '''/prepare_environment()/a\    os.system\(f\"""sed -i -e ''\"s/dict()))/dict())).cuda()/g\"'' /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py""")''' /content/stable-diffusion-webui/launch.py
+RUN sed -i -e 's/    start()/    #start()/g' /content/stable-diffusion-webui/launch.py
+RUN cd stable-diffusion-webui && python launch.py --skip-torch-cuda-test
+
+ADD https://github.com/camenduru/webui-docker/raw/main/env_patch.py /content/env_patch.py
+RUN sed -i -e '/import image_from_url_text/r /content/env_patch.py' /content/stable-diffusion-webui/modules/ui.py
+ADD https://github.com/camenduru/webui-docker/raw/main/header_patch.py /content/header_patch.py
+RUN sed -i -e '/demo:/r /content/header_patch.py' /content/stable-diffusion-webui/modules/ui.py
+
+RUN sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e "s/document.getElementsByTagName('gradio-app')\[0\].shadowRoot/!!document.getElementsByTagName('gradio-app')[0].shadowRoot ? document.getElementsByTagName('gradio-app')[0].shadowRoot : document/g" /content/stable-diffusion-webui/script.js
+RUN sed -i -e 's/                show_progress=False,/                show_progress=True,/g' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e 's/default_enabled=False/default_enabled=True/g' /content/stable-diffusion-webui/webui.py
+RUN sed -i -e 's/ outputs=\[/queue=False, &/g' /content/stable-diffusion-webui/modules/ui.py
+RUN sed -i -e 's/               queue=False,  /                /g' /content/stable-diffusion-webui/modules/ui.py
+
+RUN rm -rfv /content/stable-diffusion-webui/scripts/
+
+ADD https://github.com/camenduru/webui-docker/raw/main/shared-config.json /content/shared-config.json
+ADD https://github.com/camenduru/webui-docker/raw/main/shared-ui-config.json /content/shared-ui-config.json
+
+ADD https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.ckpt /content/stable-diffusion-webui/models/Stable-diffusion/anything-v4.5-pruned.ckpt
+ADD https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt /content/stable-diffusion-webui/models/Stable-diffusion/anything-v4.5-pruned.vae.pt
+
+RUN adduser --disabled-password --gecos '' user
+RUN chown -R user:user /content
+RUN chmod -R 777 /content
 USER user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
-
-# Pyenv
-RUN curl https://pyenv.run | bash
-ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
-
-ARG PYTHON_VERSION=3.10.12
-# Python
-RUN pyenv install $PYTHON_VERSION && \
-    pyenv global $PYTHON_VERSION && \
-    pyenv rehash && \
-    pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir \
-    datasets \
-    huggingface-hub "protobuf<4" "click<8.1"
-
-RUN git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui /content/test && \
-	git clone https://github.com/etherealxx/batchlinks-webui /content/test/extensions/batchlinks-webui  && \
-	sed -i -e 's/    start()/    #start()/g' /content/test/launch.py && \
-	cd /content/test && \
-    python launch.py --skip-torch-cuda-test && \
-    git reset --hard
-
-RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd15/resolve/main/v1-5-pruned-emaonly.ckpt -d /content/test/models/Stable-diffusion -o v1-5-pruned-emaonly.ckpt
-
-CMD cd /content/test && python launch.py --xformers --cors-allow-origins=* --api
+
+EXPOSE 7860
+
+CMD cd /content/stable-diffusion-webui && python webui.py --xformers --listen --disable-console-progressbars --enable-console-prompts --no-progressbar-hiding --ui-config-file /content/shared-ui-config.json --ui-settings-file /content/shared-config.json