We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4d587c5 commit 5575f40Copy full SHA for 5575f40
1 file changed
maxdiffusion_jax_ai_image_tpu.Dockerfile
@@ -22,7 +22,8 @@ RUN pip install -r /deps/requirements_with_jax_ai_image.txt
22
RUN if echo "$JAX_AI_IMAGE_BASEIMAGE" | grep -q "nightly"; then \
23
echo "Nightly build detected: Installing specific Flax commit and fsspec." && \
24
pip install --upgrade --force-reinstall git+https://github.com/google/flax.git@ef78d6584623511746be4824965cdef42b464583 && \
25
- pip install "fsspec==2025.10.0";
+ pip install "fsspec==2025.10.0"; \
26
+ fi
27
28
# Run the script available in JAX-AI-Image base image to generate the manifest file
29
RUN bash /jax-ai-image/generate_manifest.sh PREFIX=maxdiffusion COMMIT_HASH=$COMMIT_HASH
0 commit comments