Skip to content

GitLab

Unverified Commit c0fe9128 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

revert `keys_to_ignore` for M2M100 (#20381)

parent f2e7d270
Show whitespace changes
Inline Side-by-side
Showing with 12 additions and 1 deletion
src/transformers/models/m2m_100/modeling_m2m_100.py
View file @ c0fe9128
...@@ -1128,7 +1128,14 @@ class M2M100Decoder(M2M100PreTrainedModel): ...@@ -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): ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment