Update rosetta/Dockerfile.maxtext

Co-authored-by: Terry Kong <[email protected]>
NVIDIA · Apr 20, 2024 · c548314 · c548314
1 parent 9514f68
commit c548314
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rosetta/Dockerfile.maxtext b/rosetta/Dockerfile.maxtext
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-pax
+ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-maxtext
 ARG [email protected]
 ARG GIT_USER_NAME=NVIDIA
 # If set to "true", then will pull new local patches, the manifest.yaml and create-distribution.sh (in case it was updated).