fix grain_worker_count max value#4335

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

aireen/row_sharding_fixAI-Hypercomputer/maxtext:aireen/row_sharding_fixCopy head branch name to clipboard