Change param order for consistency

bdfe21ab · Julien Chaumond · Lysandre Debut · c536c2a4 · bdfe21ab
Commit bdfe21ab authored Nov 26, 2019 by Julien Chaumond Committed by Lysandre Debut Nov 26, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

transformers/modeling_albert.py transformers/modeling_albert.py +2 -2

No files found.
--- a/transformers/modeling_albert.py
+++ b/transformers/modeling_albert.py
@@ -571,8 +571,8 @@ class AlbertForMaskedLM(AlbertPreTrainedModel):
    def get_output_embeddings(self):
        return self.predictions.decoder

-    def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, position_ids=None, head_mask=None,
-                masked_lm_labels=None, inputs_embeds=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):
        outputs = self.albert(
            input_ids=input_ids,
            attention_mask=attention_mask,