fix: remove bucket names, update cloud build images

Author: Abhishekbhagwat

inference/trillium/JetStream-Maxtext/Llama-4-Maverick-17B-128E/README.md

@@ -275,6 +275,7 @@ The recipe uses the helm chart to run the above steps.
     helm install -f values.yaml \
     --set volumes.gcsMounts[0].bucketName=${GCS_BUCKET} \
     --set clusterName=$CLUSTER_NAME \
+    --set pathwaysDir="gs://${GCS_BUCKET}" \
     --set job.image.repository=${ARTIFACT_REGISTRY}/${JETSTREAM_MAXTEXT_IMAGE} \
     --set job.image.tag=${JETSTREAM_MAXTEXT_VERSION} \
     --set convert_hf_ckpt=false \

inference/trillium/JetStream-Maxtext/Llama-4-Maverick-17B-128E/serve-model/templates/pathways-server.yaml

@@ -23,7 +23,7 @@ spec:
   - type: ct6e-standard-4t
     topology: 8x8
     numSlices: 1
-  pathwaysDir: "gs://jacobplatin"
+  pathwaysDir: "{{ .Values.pathwaysDir }}"
   controller:
     deploymentMode: colocate_head_with_workers
     mainContainerName: jetstream

inference/trillium/JetStream-Maxtext/Llama-4-Maverick-17B-128E/values.yaml

@@ -22,7 +22,7 @@ huggingface:
     token: "hf_api_token"
 
 model:
-  name: meta-llama/Llama-4-Scout-17B-16E-Original
+  name: meta-llama/Llama-4-Maverick-17B-128E-Original
 
 job:
   image:
@@ -46,7 +46,7 @@ jetstream:
 convert_hf_ckpt: true
 
 maxtext_config:
-  load_parameters_path: "gs://jacobplatin/llama4/maverick-orbax-checkpoints/48-layer-unscanned/0/items"
+  load_parameters_path: $CHECKPOINT_TPU_UNSCANNED
   max_prefill_predict_length: 128
   max_target_length: 256
   async_checkpointing: false

inference/trillium/JetStream-Maxtext/Llama-4-Scout-17B-16E/docker/cloudbuild.yml

@@ -16,10 +16,10 @@ steps:
 - name: 'gcr.io/cloud-builders/docker'
   args:
   - 'build'
-  - '--tag=${_ARTIFACT_REGISTRY}/maxtext-jetstream:latest'
+  - '--tag=${_ARTIFACT_REGISTRY}/${_JETSTREAM_MAXTEXT_IMAGE}:${_JETSTREAM_MAXTEXT_VERSION}'
   - '--file=Dockerfile'
   - '.'
   automapSubstitutions: true
 
 images:
-- '${_ARTIFACT_REGISTRY}/maxtext-jetstream:latest'
+- '${_ARTIFACT_REGISTRY}/${_JETSTREAM_MAXTEXT_IMAGE}:${_JETSTREAM_MAXTEXT_VERSION}'