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
|
# we don't need these
|
||||||
if name.endswith((".attention.masked_bias", ".attention.bias", ".rotary_emb.inv_freq")):
|
if name.endswith((".attention.masked_bias", ".attention.bias", ".rotary_emb.inv_freq")):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if name.startswith("bert."):
|
if name.startswith("bert."):
|
||||||
name = name.removeprefix("bert.")
|
name = name.removeprefix("bert.")
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue