*.py: accidentally corrected the wrong line
This commit is contained in:
parent
154ad1236e
commit
08e2b7701f
|
|
@ -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}")
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue