Changed requirements file with JAII name & updated Dockerfile

Rohan-Bierneni · Rohan-Bierneni · commit b8de2d2961d7 · 2025-05-21T15:44:25.000Z
diff --git a/maxdiffusion_jax_ai_image_tpu.Dockerfile b/maxdiffusion_jax_ai_image_tpu.Dockerfile
@@ -16,7 +16,7 @@ WORKDIR /deps
 COPY . .
 
 # Install Maxdiffusion Jax AI Image requirements
-RUN pip install -r /deps/requirements_with_jax_stable_stack.txt
+RUN pip install -r /deps/requirements_with_jax_ai_image.txt
 
 # Run the script available in JAX-AI-Image base image to generate the manifest file
 RUN bash /jax-stable-stack/generate_manifest.sh PREFIX=maxdiffusion COMMIT_HASH=$COMMIT_HASH
diff --git a/requirements_with_jax_ai_image.txt b/requirements_with_jax_ai_image.txt
@@ -1,5 +1,5 @@
 # Requirements for Building the MaxDifussion Docker Image
-# These requirements are additional to the dependencies present in the JAX Stable Stack base image.
+# These requirements are additional to the dependencies present in the JAX AI base image.
 absl-py
 datasets
 einops==0.8.0
