From 59066327b333ee3c63160644e3c66d4fe65a1d90 Mon Sep 17 00:00:00 2001
From: Shaurya Gupta <shaurya.gup@gmail.com>
Date: Wed, 18 Sep 2024 17:13:14 -0700
Subject: [PATCH] Disable zarr3 when using single controller runtime till
 support gets added on the backend

---
 MaxText/train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/MaxText/train.py b/MaxText/train.py
index c8f0b3755..29d487313 100644
--- 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