Fix on-the-fly model conversion in Pathways by forwarding storage flags

Pass checkpoint_storage_use_ocdbt and checkpoint_storage_use_zarr3 to to_maxtext subprocess in from_pretrained to ensure compatibility with Pathways parent process.

Author: darisoy

src/maxtext/utils/model_creation_utils.py

@@ -790,6 +790,8 @@ def from_pretrained(
             "skip_jax_distributed_system=True",
             "--lazy_load_tensors=True",
             f"--simulated_cpu_devices_count={simulated_cpu_devices_count}",
+            f"checkpoint_storage_use_ocdbt={config.checkpoint_storage_use_ocdbt}",
+            f"checkpoint_storage_use_zarr3={config.checkpoint_storage_use_zarr3}",
         ]
 
         try: