add missing keys (#6719)

d17cce22 · Patrick von Platen · GitHub · a25c9fc8 · d17cce22
Unverified Commit d17cce22 authored Aug 25, 2020 by Patrick von Platen Committed by GitHub Aug 25, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/modeling_albert.py src/transformers/modeling_albert.py +1 -0

No files found.
--- a/src/transformers/modeling_albert.py
+++ b/src/transformers/modeling_albert.py
@@ -403,6 +403,7 @@ class AlbertPreTrainedModel(PreTrainedModel):

    config_class = AlbertConfig
    base_model_prefix = "albert"
+    authorized_missing_keys = [r"position_ids"]

    def _init_weights(self, module):
        """ Initialize the weights.