Merge pull request #65 from ISEEKYAN/gyhe/fix

fix raise error

Author: Thaurun

mbridge/core/bridge.py

@@ -267,6 +267,7 @@ def _save_weights_fast(
     ) -> None:
         if len(glob(os.path.join(weights_path, "*.safetensors"))) > 0:
             raise ValueError(f"The path:{weights_path} should not has safetensors files")
+        torch.distributed.barrier()
 
         def encode_filename(mcore_weight_name, *values):
             return mcore_weight_name + '--' + '--'.join(