Unverified Commit b7345d22 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

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

parent 1e00ef68
...@@ -699,6 +699,8 @@ class BertModel(BertPreTrainedModel): ...@@ -699,6 +699,8 @@ class BertModel(BertPreTrainedModel):
""" """
authorized_missing_keys = [r"position_ids"]
def __init__(self, config): def __init__(self, config):
super().__init__(config) super().__init__(config)
self.config = config self.config = config
......
...@@ -788,6 +788,8 @@ class MobileBertModel(MobileBertPreTrainedModel): ...@@ -788,6 +788,8 @@ class MobileBertModel(MobileBertPreTrainedModel):
https://arxiv.org/pdf/2004.02984.pdf https://arxiv.org/pdf/2004.02984.pdf
""" """
authorized_missing_keys = [r"position_ids"]
def __init__(self, config): def __init__(self, config):
super().__init__(config) super().__init__(config)
self.config = config self.config = config
......
...@@ -272,6 +272,7 @@ class OpenAIGPTPreTrainedModel(PreTrainedModel): ...@@ -272,6 +272,7 @@ class OpenAIGPTPreTrainedModel(PreTrainedModel):
config_class = OpenAIGPTConfig config_class = OpenAIGPTConfig
load_tf_weights = load_tf_weights_in_openai_gpt load_tf_weights = load_tf_weights_in_openai_gpt
base_model_prefix = "transformer" base_model_prefix = "transformer"
authorized_missing_keys = [r"position_ids"]
def _init_weights(self, module): def _init_weights(self, module):
""" Initialize the weights. """ Initialize the weights.
......
...@@ -375,7 +375,9 @@ XLM_INPUTS_DOCSTRING = r""" ...@@ -375,7 +375,9 @@ XLM_INPUTS_DOCSTRING = r"""
XLM_START_DOCSTRING, XLM_START_DOCSTRING,
) )
class XLMModel(XLMPreTrainedModel): class XLMModel(XLMPreTrainedModel):
def __init__(self, config): # , dico, is_encoder, with_output): authorized_missing_keys = [r"position_ids"]
def __init__(self, config):
super().__init__(config) super().__init__(config)
# encoder / decoder, output layer # encoder / decoder, output layer
......
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