"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "040c11f6dac72bc3088498aa19184da677563424"
Unverified Commit 3b20e910 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Remove version warning in pretrained BART models (#10890)

* Remove version warning in pretrained BART models

* Put it at the base model
parent 3c12e3c1
...@@ -477,6 +477,7 @@ class BartClassificationHead(nn.Module): ...@@ -477,6 +477,7 @@ class BartClassificationHead(nn.Module):
class BartPretrainedModel(PreTrainedModel): class BartPretrainedModel(PreTrainedModel):
config_class = BartConfig config_class = BartConfig
base_model_prefix = "model" base_model_prefix = "model"
_keys_to_ignore_on_load_unexpected = [r"encoder\.version", r"decoder\.version"]
def _init_weights(self, module): def _init_weights(self, module):
std = self.config.init_std std = self.config.init_std
...@@ -1200,12 +1201,7 @@ class BartModel(BartPretrainedModel): ...@@ -1200,12 +1201,7 @@ class BartModel(BartPretrainedModel):
) )
class BartForConditionalGeneration(BartPretrainedModel): class BartForConditionalGeneration(BartPretrainedModel):
base_model_prefix = "model" base_model_prefix = "model"
_keys_to_ignore_on_load_missing = [ _keys_to_ignore_on_load_missing = [r"final_logits_bias", r"lm_head\.weight"]
r"final_logits_bias",
r"encoder\.version",
r"decoder\.version",
r"lm_head\.weight",
]
def __init__(self, config: BartConfig): def __init__(self, config: BartConfig):
super().__init__(config) super().__init__(config)
......
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