Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a4db4e303208a5b7b4ce1564301d77e9b74b01d9"
Commit bdfe21ab authored by Julien Chaumond's avatar Julien Chaumond Committed by Lysandre Debut
Browse files

Change param order for consistency

parent c536c2a4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
transformers/modeling_albert.py
View file @ bdfe21ab
...@@ -571,8 +571,8 @@ class AlbertForMaskedLM(AlbertPreTrainedModel): ...@@ -571,8 +571,8 @@ class AlbertForMaskedLM(AlbertPreTrainedModel):
def get_output_embeddings(self): def get_output_embeddings(self):
return self.predictions.decoder return self.predictions.decoder
def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, position_ids=None, head_mask=None, def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, position_ids=None, head_mask=None, inputs_embeds=None,
masked_lm_labels=None, inputs_embeds=None): masked_lm_labels=None):
outputs = self.albert( outputs = self.albert(
input_ids=input_ids, input_ids=input_ids,
attention_mask=attention_mask, attention_mask=attention_mask,
......
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