Skip to content

Commit 1a52490

Browse files
committed
Add zone argument for xpk commands in multihost RL tutorials
1 parent 6b6dbc2 commit 1a52490

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

docs/tutorials/posttraining/rl_on_multi_host.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,7 @@ xpk workload create-pathways --workload ${WORKLOAD?} \
195195
--docker-image gcr.io/${PROJECT_ID?}/${CLOUD_IMAGE_NAME?} --cluster ${TPU_CLUSTER?} \
196196
--tpu-type=${TPU_TYPE?} --num-slices=1 \
197197
--project=${PROJECT_ID?} --priority=high \
198+
--zone=${ZONE?} \
198199
--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' \
199200
python3 -m src.maxtext.trainers.post_train.rl.train_rl src/maxtext/configs/post_train/rl.yml \
200201
model_name=${MODEL?} \
@@ -212,6 +213,7 @@ xpk workload create-pathways --workload ${WORKLOAD?} \
212213
--docker-image gcr.io/${PROJECT_ID?}/${CLOUD_IMAGE_NAME?} --cluster ${TPU_CLUSTER?} \
213214
--tpu-type=${TPU_TYPE?} --num-slices=1 \
214215
--project=${PROJECT_ID?} --priority=high \
216+
--zone=${ZONE?} \
215217
--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' \
216218
python3 -m src.maxtext.trainers.post_train.rl.train_rl src/maxtext/configs/post_train/rl.yml \
217219
model_name=${MODEL?} \

0 commit comments

Comments
 (0)