From 8abd7f69fcd42a5826d6911ddeabffb60dfc3fbc Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 2 Sep 2020 12:44:51 +0200 Subject: [PATCH] fix warning for position ids (#6884) --- src/transformers/modeling_electra.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/modeling_electra.py b/src/transformers/modeling_electra.py index acb2f5b416..ad49f61065 100644 --- a/src/transformers/modeling_electra.py +++ b/src/transformers/modeling_electra.py @@ -185,6 +185,7 @@ class ElectraPreTrainedModel(BertPreTrainedModel): config_class = ElectraConfig load_tf_weights = load_tf_weights_in_electra base_model_prefix = "electra" + authorized_missing_keys = [r"position_ids"] @dataclass