fix line
This commit is contained in:
parent
f5e2558f3b
commit
a0c5a0e82f
1 changed files with 1 additions and 1 deletions
|
@ -248,7 +248,7 @@ class Model:
|
|||
# we don't need these
|
||||
if name.endswith((".attention.masked_bias", ".attention.bias", ".rotary_emb.inv_freq")):
|
||||
continue
|
||||
|
||||
|
||||
if name.startswith("bert."):
|
||||
name = name.removeprefix("bert.")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue