Merge pull request #3135 from AI-Hypercomputer:sanbao/xlml

PiperOrigin-RevId: 869884433

Author: Google-ML-Automation

.github/workflows/build_and_push_docker_image.yml

@@ -131,6 +131,7 @@ jobs:
           # Add MaxText tag
           maxtext_hash=$(git rev-parse --short HEAD)
           gcloud container images add-tag "$TEMP_IMG" "$SOURCE_IMAGE:maxtext_${maxtext_hash}_${clean_date}" --quiet
+          gcloud container images add-tag "$TEMP_IMG" "$SOURCE_IMAGE:${{ inputs.image_date }}" --quiet
 
           # Add post-training dependencies tags
           for dir in tunix vllm tpu-inference; do