Dockerfile

FROM ubuntu:22.04

LABEL maintainer="Stability AI"
ENV MODEL_NAME="stabilityai/sdxl-turbo"
ENV SERVING_PORT=8080
ENV PYTHONUNBUFFERED=1
ENV SAVE_PATH="/opt/model"

COPY requirements.txt /opt/
COPY prepare_huggingface_data.py /opt/

RUN apt-get update -y \
 && apt-get install -y --no-install-recommends \
    gnupg2 \
    wget \
    python3-pip \
    python3-dev \
    python3-setuptools \
    gcc \
    && cd /usr/local/bin \
    && pip3 --no-cache-dir install --upgrade pip \
    && rm -rf /var/lib/apt/lists/* \
    && apt-get clean

# Update PATH so that the serve programs are found when the container is invoked.
ENV PATH="/opt/:${PATH}"

WORKDIR /opt

# install Python dependencies
RUN pip3 install --no-cache-dir -r requirements.txt
RUN pip3 install torch
RUN python3 prepare_huggingface_data.py

# copy the code and requirements
COPY entrypoint.py /opt/
COPY sdxl_turbo.py /opt/

EXPOSE $SERVING_PORT
ENTRYPOINT ["sh", "-c", "python3 entrypoint.py $SERVING_PORT"]

# Label image
ARG GIT_BRANCH
ARG GIT_COMMIT
LABEL org.opencontainers.image.source https://github.com/Stability-AI/marketplace-containers
LABEL org.opencontainers.image.version $GIT_BRANCH
LABEL org.opencontainers.image.revision $GIT_COMMIT
LABEL org.opencontainers.image.vendor "Stability AI"