56 lines
2.4 KiB
Docker
56 lines
2.4 KiB
Docker
FROM public.ecr.aws/ubuntu/ubuntu:22.04_stable
|
|
|
|
SHELL ["/bin/bash", "-c"]
|
|
|
|
WORKDIR /home/ubuntu/
|
|
|
|
RUN adduser --disabled-password --gecos '' ubuntu && \
|
|
apt-get update -y && \
|
|
apt-get install --no-install-recommends -y \
|
|
git libgl1 libtcmalloc-minimal4 libglib2.0-0 python3.10 python3.10-venv net-tools bc \
|
|
pkg-config file curl protobuf-compiler mesa-utils \
|
|
python3-dev libcairo2-dev libprotobuf-dev build-essential cmake wget && \
|
|
rm -rf /var/lib/apt/lists/* && \
|
|
wget -q https://github.com/peak/s5cmd/releases/download/v2.2.2/s5cmd_2.2.2_Linux-64bit.tar.gz && \
|
|
tar xzvf s5cmd_2.2.2_Linux-64bit.tar.gz && \
|
|
rm -rf s5cmd_2.2.2_Linux-64bit.tar.gz && \
|
|
mv s5cmd /usr/local/bin/
|
|
|
|
ARG ESD_COMMIT_ID
|
|
ENV ESD_COMMIT_ID=${ESD_COMMIT_ID}
|
|
ENV ON_DOCKER true
|
|
ENV LD_PRELOAD /usr/lib/x86_64-linux-gnu/libtcmalloc_minimal.so.4
|
|
ENV FULL_IMAGE true
|
|
|
|
RUN mkdir -p /home/ubuntu/conda/lib/ && \
|
|
cd /home/ubuntu/conda/lib/ && \
|
|
wget -q https://aws-gcr-solutions-us-west-2.s3.us-west-2.amazonaws.com/extension-for-stable-diffusion-on-aws/1.5.0-g5/conda/libcufft.so.10 && \
|
|
wget -q https://aws-gcr-solutions-us-west-2.s3.us-west-2.amazonaws.com/extension-for-stable-diffusion-on-aws/1.5.0-g5/conda/libcurand.so.10
|
|
|
|
COPY install_sd.sh /home/ubuntu/
|
|
RUN cd /home/ubuntu || exit 1 && \
|
|
bash install_sd.sh
|
|
|
|
COPY inference/clip_init.py /home/ubuntu/stable-diffusion-webui
|
|
RUN cd /home/ubuntu/stable-diffusion-webui/ || exit 1 && \
|
|
source venv/bin/activate python && \
|
|
python3 clip_init.py
|
|
|
|
RUN wget -qO '/home/ubuntu/stable-diffusion-webui/models/VAE-approx/vaeapprox-sdxl.pt' 'https://github.com/AUTOMATIC1111/stable-diffusion-webui/releases/download/v1.0.0-pre/vaeapprox-sdxl.pt'
|
|
|
|
RUN wget -qO '/home/ubuntu/stable-diffusion-webui/models/insightface/inswapper_128.onnx' 'https://aws-gcr-solutions-us-west-2.s3.us-west-2.amazonaws.com/extension-for-stable-diffusion-on-aws/1.5.0-g5/creator/inswapper_128.onnx'
|
|
|
|
RUN mkdir -p '/root/.u2net/' && \
|
|
wget -qO '/root/.u2net/u2net_human_seg.onnx' 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx'
|
|
|
|
RUN mkdir -p /root/.cache/huggingface/accelerate
|
|
COPY inference/default_config.yaml /root/.cache/huggingface/accelerate/
|
|
|
|
COPY inference/serve.sh /
|
|
RUN chmod +x /serve.sh
|
|
|
|
COPY inference/metrics.py /
|
|
RUN chmod +x /metrics.py
|
|
|
|
ENTRYPOINT ["bash", "/serve.sh"]
|