From 9a8860cf5d3b694170c035f6b61db72a3a59920a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sigbj=C3=B8rn=20Skj=C3=A6ret?= Date: Sat, 15 Nov 2025 14:12:39 +0100 Subject: [PATCH] convert : use all parts in safetensors index (#17286) --- convert_hf_to_gguf.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 43f74cfb14..6aaed5684b 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -189,10 +189,10 @@ class ModelBase: return tensors prefix = "model" if not self.is_mistral_format else "consolidated" - part_names: list[str] = ModelBase.get_model_part_names(self.dir_model, prefix, ".safetensors") + part_names: set[str] = set(ModelBase.get_model_part_names(self.dir_model, prefix, ".safetensors")) is_safetensors: bool = len(part_names) > 0 if not is_safetensors: - part_names = ModelBase.get_model_part_names(self.dir_model, "pytorch_model", ".bin") + part_names = set(ModelBase.get_model_part_names(self.dir_model, "pytorch_model", ".bin")) tensor_names_from_index: set[str] = set() @@ -209,6 +209,7 @@ class ModelBase: if weight_map is None or not isinstance(weight_map, dict): raise ValueError(f"Can't load 'weight_map' from {index_name!r}") tensor_names_from_index.update(weight_map.keys()) + part_names |= set(weight_map.values()) else: weight_map = {} else: