SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

revert keys_to_ignore for M2M100 (#20381)

Browse Source
This commit is contained in:
Younes Belkada
2022-11-22 13:56:23 +01:00
committed by GitHub
parent f2e7d270ec
commit c0fe912840
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/transformers/models/m2m_100/modeling_m2m_100.py
View File

@@ -1128,7 +1128,14 @@ class M2M100Decoder(M2M100PreTrainedModel):
M2M_100_START_DOCSTRING, M2M_100_START_DOCSTRING,
) )
class M2M100Model(M2M100PreTrainedModel): class M2M100Model(M2M100PreTrainedModel):
_keys_to_ignore_on_load_missing = ["encoder.embed_tokens.weight", "decoder.embed_tokens.weight"] _keys_to_ignore_on_load_missing = [
"encoder.embed_tokens.weight",
"decoder.embed_tokens.weight",
"encoder.embed_positions.weights",
"encoder.embed_positions.bias",
"decoder.embed_positions.weights",
"decoder.embed_positions.bias",
]
def __init__(self, config: M2M100Config): def __init__(self, config: M2M100Config):
super().__init__(config) super().__init__(config)
@@ -1248,6 +1255,10 @@ class M2M100ForConditionalGeneration(M2M100PreTrainedModel):
r"lm_head.weight", r"lm_head.weight",
r"encoder.embed_tokens.weight", r"encoder.embed_tokens.weight",
r"decoder.embed_tokens.weight", r"decoder.embed_tokens.weight",
r"encoder.embed_positions.weights",
r"encoder.embed_positions.bias",
r"decoder.embed_positions.weights",
r"decoder.embed_positions.bias",
] ]
def __init__(self, config: M2M100Config): def __init__(self, config: M2M100Config):