point

AI-Hypercomputer · Jul 25, 2024 · 16d6be5 · 16d6be5
1 parent 1197bed
commit 16d6be5
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/maxdiffusion_jax_stable_stack_tpu.Dockerfile b/maxdiffusion_jax_stable_stack_tpu.Dockerfile
@@ -16,16 +16,16 @@ WORKDIR /deps
 COPY . .
 RUN ls .
 
-# ARG MAXDIFFUSION_REQUIREMENTS_FILE
+ARG MAXDIFFUSION_REQUIREMENTS_FILE
 
-# # Install Maxdiffusion requirements
-# RUN if [ ! -z "${MAXDIFFUSION_REQUIREMENTS_FILE}" ]; then \
-#         echo "Using MaxDiffusion requirements: ${MAXDIFFUSION_REQUIREMENTS_FILE}" && \
-#         pip install -r /deps/${MAXDIFFUSION_REQUIREMENTS_FILE}; \
-#     fi
+# Install Maxdiffusion requirements
+RUN if [ ! -z "${MAXDIFFUSION_REQUIREMENTS_FILE}" ]; then \
+        echo "Using MaxDiffusion requirements: ${MAXDIFFUSION_REQUIREMENTS_FILE}" && \
+        pip install -r /deps/${MAXDIFFUSION_REQUIREMENTS_FILE}; \
+    fi
 
-# # Install MaxDiffusion
-# RUN pip install .
+# Install MaxDiffusion
+RUN pip install .
 
 # Run the script available in JAX-Stable-Stack base image to generate the manifest file
 RUN bash /generate_manifest.sh PREFIX=maxdiffusion COMMIT_HASH=$COMMIT_HASH