From 2e4ff37929a27613788fd11f60c547686f8729d9 Mon Sep 17 00:00:00 2001 From: guanyouhe Date: Fri, 9 Jan 2026 20:29:05 +0800 Subject: [PATCH] fix raise --- mbridge/core/bridge.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mbridge/core/bridge.py b/mbridge/core/bridge.py index 6ad5503..085413a 100644 --- a/mbridge/core/bridge.py +++ b/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(