File tree Expand file tree Collapse file tree
docs/tutorials/posttraining Expand file tree Collapse file tree Original file line number Diff line number Diff 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' \
199200python3 -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' \
216218python3 -m src.maxtext.trainers.post_train.rl.train_rl src/maxtext/configs/post_train/rl.yml \
217219 model_name=${MODEL?} \
You can’t perform that action at this time.
0 commit comments