From ab7a7ce0119532a4b3bec5f79f1ae9db015a1301 Mon Sep 17 00:00:00 2001 From: Branden Vandermoon Date: Mon, 15 Jun 2026 18:37:13 +0000 Subject: [PATCH] Update outdated rollout parallelism comment --- src/maxtext/configs/post_train/rl.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/maxtext/configs/post_train/rl.yml b/src/maxtext/configs/post_train/rl.yml index 9ba55c296f..8ce27d4c34 100644 --- a/src/maxtext/configs/post_train/rl.yml +++ b/src/maxtext/configs/post_train/rl.yml @@ -23,8 +23,7 @@ sampler_devices_fraction: 0.5 chips_per_vm: 4 # depends on hardware, for v5p this is 4 num_trainer_slices: -1 num_samplers_slices: -1 -# Only specify rollout_data_parallelism when you would like to use more than one model -# replicas in rollout. If not specified, rollout_tensor_parallelism will be auto-determined. +# If rollout_data_parallelism is not specified, it will be auto-determined. rollout_data_parallelism: -1 rollout_tensor_parallelism: 1 rollout_expert_parallelism: 1