Dockerfile

#
# Command to build the image:
#   sudo docker build -t sd .
#
# Example command to start the container (weights and result are saved in the current directory):
#   sudo docker run -t -i --init -v .:/current sd --models-path /current --output /current/result.png --steps 1 --turbo
#

ARG UBUNTU_VERSION=22.04
FROM ubuntu:$UBUNTU_VERSION as build
RUN apt-get update && apt-get install -y build-essential git cmake
WORKDIR /
RUN git clone https://github.com/google/XNNPACK.git \
    && cd XNNPACK \
    && git checkout 1c8ee1b68f3a3e0847ec3c53c186c5909fa3fbd3 \
    && mkdir build \
    && cd build \
    && cmake -DXNNPACK_BUILD_TESTS=OFF -DXNNPACK_BUILD_BENCHMARKS=OFF .. \
    && cmake --build . --config Release
WORKDIR OnnxStream
COPY src .
RUN mkdir build \
    && cd build \
    && cmake -DMAX_SPEED=ON -DOS_LLM=OFF -DOS_CUDA=OFF -DXNNPACK_DIR=/XNNPACK .. \
    && cmake --build . --config Release
FROM ubuntu:$UBUNTU_VERSION as runtime
RUN apt-get update && apt-get install -y curl
COPY --from=build /OnnxStream/build/sd /sd
ENTRYPOINT [ "/sd" ]