Merge pull request #1159 from AI-Hypercomputer:vivianrwu/maxengine_do…

…cker PiperOrigin-RevId: 719049135
AI-Hypercomputer · Jan 24, 2025 · 9e91ccb · 9e91ccb
2 parents ce378ec + a57b710
commit 9e91ccb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/inference/maxengine_server/Dockerfile b/inference/maxengine_server/Dockerfile
@@ -19,6 +19,7 @@ FROM ubuntu:22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV MAXTEXT_VERSION=main
+ENV JETSTREAM_VERSION=main
 
 RUN apt -y update && apt install -y --no-install-recommends \
     ca-certificates \
@@ -37,6 +38,7 @@ git checkout ${MAXTEXT_VERSION} && \
 bash setup.sh
 
 RUN cd /JetStream && \
+git checkout ${JETSTREAM_VERSION} && \
 pip install -e .
 
 COPY maxengine_server_entrypoint.sh /usr/bin/