From bb259aeee79c14c59476cd4009d85514479a1c8e Mon Sep 17 00:00:00 2001 From: xiaotaoliu Date: Mon, 9 Feb 2026 12:10:52 +0800 Subject: [PATCH] fix save weight fast Nonetype error --- mbridge/core/bridge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mbridge/core/bridge.py b/mbridge/core/bridge.py index ed6fb2b..7a18420 100644 --- a/mbridge/core/bridge.py +++ b/mbridge/core/bridge.py @@ -380,7 +380,7 @@ def load_file(file_tuple): self.safetensor_io.save_tmp_weight(hf_name, hf_param, weights_path) else: # gather tp - if w_files[0][4] > 0: + if w_files[0][4] is not None and w_files[0][4] > 0: assert len(w_files) == w_files[0][3] params = [load_file(w_file) for w_file in w_files] infer_params = self._weight_merge_across_tp(w_name, params, params[0])