Skip to content

GitLab

Unverified Commit 92cfe8b0 authored by Arthur's avatar Arthur Committed by GitHub
Browse files

Remove `authorized_missing_keys`in favor of _keys_to_ignore_on_load_missing (#20228)

parent 2d920010
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/convbert/modeling_convbert.py
View file @ 92cfe8b0
...@@ -246,8 +246,8 @@ class ConvBertPreTrainedModel(PreTrainedModel): ...@@ -246,8 +246,8 @@ class ConvBertPreTrainedModel(PreTrainedModel):
load_tf_weights = load_tf_weights_in_convbert load_tf_weights = load_tf_weights_in_convbert
base_model_prefix = "convbert" base_model_prefix = "convbert"
supports_gradient_checkpointing = True supports_gradient_checkpointing = True
authorized_missing_keys = [r"position_ids"] _keys_to_ignore_on_load_missing = [r"position_ids"]
authorized_unexpected_keys = [r"convbert.embeddings_project.weight", r"convbert.embeddings_project.bias"] _keys_to_ignore_on_load_unexpected = [r"convbert.embeddings_project.weight", r"convbert.embeddings_project.bias"]
def _init_weights(self, module): def _init_weights(self, module):
"""Initialize the weights""" """Initialize the weights"""
......
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