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 @@ -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' \
172171python3 -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' \
190188python3 -m maxtext.trainers.post_train.rl.train_rl \
191189 model_name=${MODEL?} \
You can’t perform that action at this time.
0 commit comments