fix warning (#9231)

6b034309 · Patrick von Platen · GitHub · a4b21cdd · 6b034309
Unverified Commit 6b034309 authored Dec 21, 2020 by Patrick von Platen Committed by GitHub Dec 21, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/transformers/models/t5/modeling_t5.py src/transformers/models/t5/modeling_t5.py +4 -0

No files found.
--- a/src/transformers/models/t5/modeling_t5.py
+++ b/src/transformers/models/t5/modeling_t5.py
@@ -1127,6 +1127,8 @@ class T5Model(T5PreTrainedModel):
    _keys_to_ignore_on_load_missing = [
        r"encoder\.embed_tokens\.weight",
        r"decoder\.embed_tokens\.weight",
+    ]
+    _keys_to_ignore_on_load_unexpected = [
        r"decoder\.block\.0\.layer\.1\.EncDecAttention\.relative_attention_bias\.weight",
    ]

@@ -1300,6 +1302,8 @@ class T5ForConditionalGeneration(T5PreTrainedModel):
        r"encoder\.embed_tokens\.weight",
        r"decoder\.embed_tokens\.weight",
        r"lm_head\.weight",
+    ]
+    _keys_to_ignore_on_load_unexpected = [
        r"decoder\.block\.0\.layer\.1\.EncDecAttention\.relative_attention_bias\.weight",
    ]