fix line
This commit is contained in:
parent
f5e2558f3b
commit
a0c5a0e82f
|
|
@ -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…
Reference in New Issue