Local Image Name

AI-Hypercomputer · Feb 26, 2024 · 0e36803 · 0e36803
1 parent f2cd92c
commit 0e36803
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/BuildDockerImages.yml b/.github/workflows/BuildDockerImages.yml
@@ -36,7 +36,7 @@ jobs:
         project=tpu-prod-env-multipod
         local_image_name=maxtext_local_jax_stable
         cloud_image_name=maxtext_jax_stable
-        bash docker_build_dependency_image.sh $local_image_name MODE=stable &&
+        bash docker_build_dependency_image.sh LOCAL_IMAGE_NAME=$local_image_name MODE=stable &&
         docker tag ${local_image_name} gcr.io/$project/${cloud_image_name}:latest
         docker push gcr.io/$project/${cloud_image_name}:latest