Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "5603f78fc46bd117e3f25cc8842eb08046bbff4e"
Unverified Commit 6b034309 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix warning (#9231)

parent a4b21cdd
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/transformers/models/t5/modeling_t5.py
View file @ 6b034309
...@@ -1127,6 +1127,8 @@ class T5Model(T5PreTrainedModel): ...@@ -1127,6 +1127,8 @@ class T5Model(T5PreTrainedModel):
_keys_to_ignore_on_load_missing = [ _keys_to_ignore_on_load_missing = [
r"encoder\.embed_tokens\.weight", r"encoder\.embed_tokens\.weight",
r"decoder\.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", r"decoder\.block\.0\.layer\.1\.EncDecAttention\.relative_attention_bias\.weight",
] ]
...@@ -1300,6 +1302,8 @@ class T5ForConditionalGeneration(T5PreTrainedModel): ...@@ -1300,6 +1302,8 @@ class T5ForConditionalGeneration(T5PreTrainedModel):
r"encoder\.embed_tokens\.weight", r"encoder\.embed_tokens\.weight",
r"decoder\.embed_tokens\.weight", r"decoder\.embed_tokens\.weight",
r"lm_head\.weight", r"lm_head\.weight",
]
_keys_to_ignore_on_load_unexpected = [
r"decoder\.block\.0\.layer\.1\.EncDecAttention\.relative_attention_bias\.weight", r"decoder\.block\.0\.layer\.1\.EncDecAttention\.relative_attention_bias\.weight",
] ]
......
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