Skip to content

Migrated naming of jax stable stack -> jax ai image#177

Merged
parambole merged 1 commit intomainfrom
rbierneni-jaii-migration
May 21, 2025
Merged

Migrated naming of jax stable stack -> jax ai image#177
parambole merged 1 commit intomainfrom
rbierneni-jaii-migration

Conversation

@Rohan-Bierneni
Copy link
Copy Markdown
Collaborator

Updated the Docker file for JAX AI Image + MaxDiffusion build and updated the getting started tutorials to use JAX AI Image instead of JAX Stable Stack. Also, updated the daily image builds to use the new Artifact Registry repo and added a new mode "jax_ai_image" which performs the exact same as the mode "stable_stack". As of now both modes are supported but in a follow up pr, will deprecate mode "stable_stack".

@Rohan-Bierneni Rohan-Bierneni self-assigned this May 20, 2025
Comment thread docker_build_dependency_image.sh
Comment thread docs/getting_started/run_maxdiffusion_via_xpk.md
parambole
parambole previously approved these changes May 20, 2025
Copy link
Copy Markdown
Collaborator

@parambole parambole left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM with nits. Thank you for making this change.

Changed requirements file with JAII name & updated Dockerfile

Resolved comments. Added deprecation warning and notes
@Rohan-Bierneni Rohan-Bierneni force-pushed the rbierneni-jaii-migration branch from 0a0eb9c to 168c086 Compare May 21, 2025 15:54
Copy link
Copy Markdown
Collaborator

@parambole parambole left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Rohan-Bierneni Rohan-Bierneni marked this pull request as ready for review May 21, 2025 17:55
@parambole parambole merged commit 4a8155e into main May 21, 2025
2 of 3 checks passed
hx89 pushed a commit to hx89/maxdiffusion that referenced this pull request May 28, 2025
Changed requirements file with JAII name & updated Dockerfile

Resolved comments. Added deprecation warning and notes
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants