protogen-web-ui / Dockerfile.old
darkstorm2150's picture
Rename Dockerfile to Dockerfile.old
54b18bd verified
raw
history blame
2.1 kB
# Use a smaller base image if possible, but maintain CUDA compatibility
FROM nvidia/cuda:11.7.1-base-ubuntu22.04 AS base
ENV DEBIAN_FRONTEND=noninteractive
# Create user and directories early to leverage caching
RUN adduser --disabled-password --gecos '' user && \
mkdir -p /content && \
chown -R user:user /content
# Install system dependencies efficiently
RUN apt-get update && \
apt-get install -y --no-install-recommends \
libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && \
rm -rf /var/lib/apt/lists/*
# Switch to user after system setup
WORKDIR /content
USER user
# Combine pip installations into a single layer
RUN pip3 install --upgrade pip==23.3.1 && \
pip install --no-cache-dir \
torchmetrics==0.11.4 \
numexpr \
httpx==0.24.1 \
pytorch_lightning==1.7.6 \
https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl \
--pre triton
# Clone repository and apply patches in a single step
RUN rm -rf stable-diffusion-webui && \
git clone -b v1.6 https://github.com/camenduru/stable-diffusion-webui && \
cd stable-diffusion-webui && \
sed -i 's/dict()))/dict())).cuda()/g' repositories/stable-diffusion-stability-ai/ldm/util.py && \
sed -i '/prepare_environment()/a os.system("sed -i -e '\''s/dict()))/dict())).cuda()/g'\'' repositories/stable-diffusion-stability-ai/ldm/util.py")' launch.py && \
sed -i 's/start()/#start()/g' launch.py && \
echo "fastapi==0.90.0" >> requirements_versions.txt && \
python launch.py --skip-torch-cuda-test
# Add Model
ADD --chown=user https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors /content/stable-diffusion-webui/models/Stable-diffusion/OpenGen%20v1.0.safetensors
EXPOSE 7860
CMD ["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"]