Spaces:
Runtime error
Runtime error
| FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND noninteractive | |
| ENV PYTHONUNBUFFERED=1 | |
| ENV PIP_DISABLE_PIP_VERSION_CHECK=1 | |
| ENV PIP_NO_CACHE_DIR=1 | |
| # OS setup | |
| RUN apt-get update -y \ | |
| && apt-get upgrade -y \ | |
| && apt-get install -y \ | |
| libgl1 \ | |
| libglib2.0-0 \ | |
| curl \ | |
| vim \ | |
| wget \ | |
| git \ | |
| git-lfs \ | |
| tzdata \ | |
| bash \ | |
| ca-certificates \ | |
| libreadline8 \ | |
| bzip2 \ | |
| psmisc \ | |
| procps \ | |
| netbase \ | |
| openssh-client \ | |
| libsqlite3-dev \ | |
| python3-pip \ | |
| python3-venv \ | |
| python-is-python3 \ | |
| build-essential \ | |
| libssl-dev \ | |
| libffi-dev \ | |
| aria2 \ | |
| \ | |
| && pip3 install --upgrade pip \ | |
| \ | |
| && git lfs install \ | |
| \ | |
| && apt-get clean autoclean \ | |
| && apt-get autoremove --yes \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # OS timezone setting (UTC) | |
| RUN echo "UTC" > /etc/timezone | |
| ENV TZ=UTC | |
| # Poetry for Python packages | |
| RUN curl -sSL https://install.python-poetry.org | POETRY_HOME=/usr/local/poetry python3 - --yes \ | |
| && ln -s /usr/local/poetry/bin/poetry /usr/bin/poetry \ | |
| \ | |
| && poetry config virtualenvs.create false \ | |
| && poetry config virtualenvs.in-project false | |
| # Create non-root user | |
| ENV ENV="/etc/profile" | |
| RUN adduser --disabled-password --gecos '' user && \ | |
| mkdir -p /app && \ | |
| chown -R user:user /app && \ | |
| printf "\n. /etc/profile\n" >> /home/user/.profile \ | |
| printf "\n. /etc/profile\n" >> /home/user/.bashrc | |
| # Sets up virtualenv for dependencies | |
| ENV VIRTUAL_ENV="/opt/venv" | |
| ENV VIRTUAL_ENV_DISABLE_PROMPT=1 | |
| ENV POETRY_ACTIVE=1 | |
| ENV PATH="$VIRTUAL_ENV/bin:$PATH" | |
| RUN echo "export PATH=$PATH" >> /home/user/.bashrc \ | |
| && python3 -m venv $VIRTUAL_ENV \ | |
| && /opt/venv/bin/pip install --upgrade --no-cache-dir pip \ | |
| && pip3 install httpx==0.24.1 \ | |
| && chown -R user:user /opt/venv | |
| # Run as non-root user | |
| USER user | |
| WORKDIR /app | |
| # Installation of basic Python dependencies specified in pyproject.toml | |
| COPY --chown=user:user pyproject.toml poetry.lock /app/ | |
| RUN poetry install | |
| # AUTOMATIC1111' WebUI | |
| RUN git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui /app/stable-diffusion-webui \ | |
| && (cd /app/stable-diffusion-webui && git checkout a9fed7c364061ae6efb37f797b6b522cb3cf7aa2) | |
| # Deforum extension | |
| RUN git clone https://github.com/deforum-art/deforum-for-automatic1111-webui /app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui \ | |
| && (cd /app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui && git checkout 2366bfdb47c226df0d14e712445414e459febad3) | |
| # Images Browser WebUI extension | |
| RUN git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser \ | |
| && (cd /app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser && git checkout a42c7a30181636a05815e62426d5eff4d3340529) | |
| # CiviTAI Browser WebUI extension | |
| RUN git clone https://github.com/Vetchems/sd-civitai-browser /app/stable-diffusion-webui/extensions/sd-civitai-browser \ | |
| && (cd /app/stable-diffusion-webui/extensions/sd-civitai-browser && git checkout b25a5daf7df3f6340d3e243d533228d8ade5288d) | |
| # Additional Networks WebUI extension | |
| RUN git clone https://github.com/kohya-ss/sd-webui-additional-networks /app/stable-diffusion-webui/extensions/sd-webui-additional-networks \ | |
| && (cd /app/stable-diffusion-webui/extensions/sd-webui-additional-networks && git checkout d2758b6c8e2e8e956865a87b31fd74d3d7c010cb) \ | |
| && mkdir -p /app/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/LoRA | |
| # ControlNet WebUI extension | |
| RUN git clone https://github.com/Mikubill/sd-webui-controlnet /app/stable-diffusion-webui/extensions/sd-webui-controlnet \ | |
| && (cd /app/stable-diffusion-webui/extensions/sd-webui-controlnet && git checkout 274dd5df217a03e059e9cf052447aece81bbd1cf) \ | |
| && mkdir -p /app/stable-diffusion-webui/models/ControlNet | |
| # Prepare WebUI environment | |
| WORKDIR /app/stable-diffusion-webui | |
| RUN /opt/venv/bin/python launch.py --exit --skip-torch-cuda-test --xformers | |
| # Patch WebUI | |
| RUN sed -i -e 's/ show_progress=False,/ show_progress=True,/g' modules/ui.py | |
| RUN sed -i -e 's/shared.demo.launch/shared.demo.queue().launch/g' webui.py | |
| RUN sed -i -e 's/ outputs=\[/queue=False, &/g' modules/ui.py | |
| RUN sed -i -e 's/ queue=False, / /g' modules/ui.py | |
| # Copy startup scripts | |
| COPY --chown=user:user run.py on_start.sh config.json ui-config.json shared-config.json shared-ui-config.json header_patch.py /app/stable-diffusion-webui/ | |
| RUN chmod +x on_start.sh | |
| EXPOSE 7860 | |
| CMD ["/opt/venv/bin/python", "run.py", "--listen", "--ui-config-file", "ui-config.json", "--ui-settings-file", "config.json", "--disable-console-progressbars", "--cors-allow-origins", "huggingface.co,hf.space", "--no-progressbar-hiding", "--enable-console-prompts", "--no-download-sd-model", "--api", "--skip-version-check"] | |