From 6b034309ca4ca2ec6e5c3cacda92a448fa10b921 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 21 Dec 2020 10:41:34 +0100 Subject: [PATCH] fix warning (#9231) --- src/transformers/models/t5/modeling_t5.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/transformers/models/t5/modeling_t5.py b/src/transformers/models/t5/modeling_t5.py index 491ece4d99..0ce2be3c62 100644 --- 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", ]