Update merge_utils.py

merge_utils.py

@@ -40,5 +40,6 @@ def execute_mergekit(config_dict, out_path, hf_token):
         low_cpu_mem=True,
         copy_tokenizer=True,
         lazy_unpickle=True
+        max_shard_size=int(shard_gb * 1024 * 1024 * 1024)
     )
     return True