Skip to content

GitLab

Unverified Commit 8d464374 authored by Raphaël Bournhonesque's avatar Raphaël Bournhonesque Committed by GitHub
Browse files

Disable missing weight warning (#7282)

parent 8ff88d25
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/modeling_roberta.py
View file @ 8d464374
...@@ -303,6 +303,7 @@ class RobertaForCausalLM(BertPreTrainedModel): ...@@ -303,6 +303,7 @@ class RobertaForCausalLM(BertPreTrainedModel):
class RobertaForMaskedLM(BertPreTrainedModel): class RobertaForMaskedLM(BertPreTrainedModel):
config_class = RobertaConfig config_class = RobertaConfig
base_model_prefix = "roberta" base_model_prefix = "roberta"
authorized_missing_keys = [r"position_ids", r"lm_head\.decoder\.bias"]
def __init__(self, config): def __init__(self, config):
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