diff --git a/container-images/musa/Containerfile b/container-images/musa/Containerfile index 8a4480f09..aa13f8949 100644 --- a/container-images/musa/Containerfile +++ b/container-images/musa/Containerfile @@ -1,13 +1,13 @@ -ARG VERSION=rc4.0.1 +ARG VERSION=rc4.2.0 ARG UBUNTU_VERSION=22.04 # Base image with MUSA for compilation -FROM docker.io/mthreads/musa:${VERSION}-mudnn-devel-ubuntu${UBUNTU_VERSION} AS builder +FROM docker.io/mthreads/musa:${VERSION}-devel-ubuntu${UBUNTU_VERSION}-amd64 AS builder COPY --chmod=755 ../scripts /usr/bin RUN build_llama_and_whisper.sh "musa" # Final runtime image -FROM docker.io/mthreads/musa:${VERSION}-mudnn-runtime-ubuntu${UBUNTU_VERSION} +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