Skip to content

Commit 03a192b

Browse files
Merge pull request #3986 from AI-Hypercomputer:doc_change_multihost
PiperOrigin-RevId: 922857041
2 parents 56bf0f5 + 7c976c2 commit 03a192b

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

docs/tutorials/posttraining/rl_on_multi_host.md

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ placeholders with your actual values.
7979
# -- Model configuration --
8080
# The MaxText model name. See `src/maxtext/configs/types.py` for `ModelName` for a
8181
# full list of supported models.
82-
export MODEL=<MODEL_NAME> # e.g. 'llama3.1-70b-Instruct'
82+
export MODEL=<MODEL_NAME> # e.g. 'llama3.1-70b-Instruct' # replace with another model from src/maxtext/configs/types.py if needed
8383

8484
# Your Hugging Face access token. Required to download gated models like Llama.
8585
# You can generate one at https://huggingface.co/settings/tokens.
@@ -167,7 +167,6 @@ xpk workload create-pathways --workload ${RUN_NAME?} \
167167
--docker-image ${DOCKER_IMAGE?} --cluster ${GKE_CLUSTER?} \
168168
--tpu-type=${TPU_TYPE?} --num-slices=1 \
169169
--project=${PROJECT_ID?} --priority=high \
170-
--zone=${ZONE?} \
171170
--command "HF_TOKEN=${HF_TOKEN?} TF_CPP_MIN_LOG_LEVEL=0 JAX_PLATFORMS=proxy JAX_BACKEND_TARGET=grpc://127.0.0.1:29000 ENABLE_PATHWAYS_PERSISTENCE='1' \
172171
python3 -m maxtext.trainers.post_train.rl.train_rl \
173172
model_name=${MODEL?} \
@@ -185,7 +184,6 @@ xpk workload create-pathways --workload ${RUN_NAME?} \
185184
--docker-image ${DOCKER_IMAGE?} --cluster ${GKE_CLUSTER?} \
186185
--tpu-type=${TPU_TYPE?} --num-slices=1 \
187186
--project=${PROJECT_ID?} --priority=high \
188-
--zone=${ZONE?} \
189187
--command "HF_TOKEN=${HF_TOKEN?} TF_CPP_MIN_LOG_LEVEL=0 JAX_PLATFORMS=proxy JAX_BACKEND_TARGET=grpc://127.0.0.1:29000 ENABLE_PATHWAYS_PERSISTENCE='1' \
190188
python3 -m maxtext.trainers.post_train.rl.train_rl \
191189
model_name=${MODEL?} \

0 commit comments

Comments
 (0)