update

Author: hengtaoguo

tests/end_to_end/tpu/qwen/moe/qwen3-omni-30b-a3b/2_test_qwen3_omni_30b_a3b.sh

@@ -17,27 +17,29 @@ set -ex
 export MODEL_NAME="${MODEL_NAME:-qwen3-omni-30b-a3b}"
 export TOKENIZER_PATH="${TOKENIZER_PATH:-Qwen/Qwen3-Omni-30B-A3B-Instruct}"
 
-# Base output path where the MaxText checkpoint from Step 1 was written.
-export BASE_OUTPUT_PATH="${BASE_OUTPUT_PATH:-gs://your-gcs-bucket/qwen3-omni-30b-a3b_maxtext_ckpt}"
+if [ -z "${BASE_OUTPUT_PATH}" ]; then
+  # Non-Googlers please remember to point `BASE_OUTPUT_PATH` to GCS buckets that you own, this script uses internal buckets for testing.
+  # this bucket will store all the files generated by MaxText during a run
+  export BASE_OUTPUT_PATH=gs://runner-maxtext-logs/$(date +%Y-%m-%d-%H-%M)
+  echo "BASE_OUTPUT_PATH is not set"
+fi
+BASE_OUTPUT_PATH=${BASE_OUTPUT_PATH%/}
+echo using BASE_OUTPUT_PATH = ${BASE_OUTPUT_PATH}
 
 if [ -z "${HF_TOKEN}" ]; then
   echo "Error: HF_TOKEN environment variable is not set. Please export your Hugging Face token."
   echo "Example: export HF_TOKEN=hf_..."
   exit 1
 fi
 
-# Strip trailing slash from base path to avoid malformed URIs
-BASE_OUTPUT_PATH=${BASE_OUTPUT_PATH%/}
-echo "Using BASE_OUTPUT_PATH = ${BASE_OUTPUT_PATH}"
-
-UNSCANNED_CKPT_PATH=${BASE_OUTPUT_PATH}/unscanned/0/items
+UNSCANNED_CKPT_PATH=gs://maxtext-model-checkpoints/qwen3-omni-30b-a3b/unscanned/0/items
 
 # ---
 # Step 2a: Multimodal Decode — text + image
 # Uses a test image from the repo assets.
 # max_prefill_predict_length accounts for image tokens (~256) + text prompt tokens.
 # ---
-python3 -m maxtext.inference.decode src/maxtext/configs/base.yml \
+python3 -m maxtext.inference.decode ${MAXTEXT_CONFIGS_DIR:-${MAXTEXT_REPO_ROOT:-$PWD}/src/maxtext/configs}/base.yml \
     model_name=${MODEL_NAME} \
     tokenizer_path=${TOKENIZER_PATH} \
     tokenizer_type=huggingface \
@@ -62,7 +64,7 @@ python3 -m maxtext.inference.decode src/maxtext/configs/base.yml \
 # max_prefill_predict_length is set higher to accommodate video frame tokens (~1126)
 # plus audio tokens (~77) plus text prompt tokens.
 # ---
-python3 -m maxtext.inference.decode src/maxtext/configs/base.yml \
+python3 -m maxtext.inference.decode ${MAXTEXT_CONFIGS_DIR:-${MAXTEXT_REPO_ROOT:-$PWD}/src/maxtext/configs}/base.yml \
     model_name=${MODEL_NAME} \
     tokenizer_path=${TOKENIZER_PATH} \
     tokenizer_type=huggingface \