Disable zarr3 when using single controller runtime

till support gets added on the backend
google · Sep 19, 2024 · 5906632 · 5906632
1 parent 817bcda
commit 5906632
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/MaxText/train.py b/MaxText/train.py
@@ -495,8 +495,6 @@ def setup_mesh_and_model(config):
         logger,
         use_ocdbt,
         use_zarr3,
-        use_ocdbt,
-        use_zarr3,
     )
 
   return init_rng, writer, checkpoint_manager, mesh, model, learning_rate_schedule, tx