diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index a499c0ef2..170be0233 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1186,7 +1186,7 @@ class RefactModel(Model): data = data_torch.squeeze().numpy() # map tensor names - new_name = tensor_map.get_name(name, try_suffixes=(".weight", ".bias")) + new_name = tensor_map.get_name(name, try_suffixes=(".weight",)) if new_name is None: raise ValueError(f"Can not map tensor {name!r}") @@ -2181,7 +2181,7 @@ class PlamoModel(Model): continue # map tensor names - new_name = tensor_map.get_name(name, try_suffixes=(".weight",)) + new_name = tensor_map.get_name(name, try_suffixes=(".weight", ".bias")) if new_name is None: raise ValueError(f"Can not map tensor {name!r}")