*.py: accidentally corrected the wrong line

This commit is contained in:
brian khuu 2024-05-02 13:35:54 +10:00
parent 154ad1236e
commit 08e2b7701f

View file

@ -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}")