ARG VERSION=rc4.2.0
ARG UBUNTU_VERSION=22.04
# Base image with MUSA for compilation
FROM docker.io/mthreads/musa:${VERSION}-devel-ubuntu${UBUNTU_VERSION}-amd64 AS builder

COPY container-images/scripts/build_llama_and_whisper.sh \
     container-images/scripts/lib.sh \
     /src/
WORKDIR /src/
RUN ./build_llama_and_whisper.sh musa

# Final runtime image
FROM docker.io/mthreads/musa:${VERSION}-runtime-ubuntu${UBUNTU_VERSION}-amd64

# Copy the entire installation directory from the builder
COPY --from=builder /tmp/install /usr
# pip install . --prefix=/tmp/install will install the wheel in /tmp/install/local/...

RUN apt-get update && apt-get install -y python-is-python3 && apt-get clean && \
    rm -rf /var/lib/apt/lists/*

ENTRYPOINT []
