"backend/apps/vscode:/vscode.git/clone" did not exist on "54420f530001a038286f162143e7239a92a97ddb"
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):
class BartPretrainedModel(PreTrainedModel):
config_class = BartConfig
base_model_prefix = "model"
_keys_to_ignore_on_load_unexpected = [r"encoder\.version", r"decoder\.version"]
def _init_weights(self, module):
std = self.config.init_std
......@@ -1200,12 +1201,7 @@ class BartModel(BartPretrainedModel):
)
class BartForConditionalGeneration(BartPretrainedModel):
base_model_prefix = "model"
_keys_to_ignore_on_load_missing = [
r"final_logits_bias",
r"encoder\.version",
r"decoder\.version",
r"lm_head\.weight",
]
_keys_to_ignore_on_load_missing = [r"final_logits_bias", r"lm_head\.weight"]
def __init__(self, config: BartConfig):
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