[fix] no warning for position_ids buffer (#6063)

This commit is contained in:
Sam Shleifer
2020-07-27 20:00:44 -04:00
committed by GitHub
parent 1e00ef681d
commit b7345d22d0
4 changed files with 8 additions and 1 deletions

View File

@@ -788,6 +788,8 @@ class MobileBertModel(MobileBertPreTrainedModel):
https://arxiv.org/pdf/2004.02984.pdf
"""
authorized_missing_keys = [r"position_ids"]
def __init__(self, config):
super().__init__(config)
self.config = config