Update microbenchmarks_llama2-70b-trillium_2x4.sh

AI-Hypercomputer · Feb 8, 2025 · b556197 · b556197
1 parent 045fc14
commit b556197
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MaxText/inference_mlperf/trillium/microbenchmarks_llama2-70b-trillium_2x4.sh b/MaxText/inference_mlperf/trillium/microbenchmarks_llama2-70b-trillium_2x4.sh
@@ -58,7 +58,7 @@ echo "LIBTPU_INIT_ARGS:${LIBTPU_INIT_ARGS}"
 echo "XLA_FLAGS:${XLA_FLAGS}"
 echo
 export TOKENIZER_PATH=/home/${USER}/maxtext/assets/tokenizer.llama2
-export LOAD_PARAMETERS_PATH=gs://msingh-bkt/checkpoints/quant_llama2-70b-chat/prod/int8_
+export LOAD_PARAMETERS_PATH=gs://${USER}-bkt/checkpoints/quant_llama2-70b-chat/prod/int8_
 export MAX_PREFILL_PREDICT_LENGTH=1024
 export MAX_TARGET_LENGTH=2048
 export MODEL_NAME=llama2-70b