*.py: accidentally corrected the wrong line

ggerganov · May 2, 2024 · 08e2b77 · 08e2b77
1 parent 154ad12
commit 08e2b77
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -1186,7 +1186,7 @@ def write_tensors(self):
             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 @@ def write_tensors(self):
                 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}")