Patch ALBERT with heads in TensorFlow

2708b44e · Lysandre · 1abd53b1 · 2708b44e
Commit 2708b44e authored Feb 19, 2020 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 8 deletions

src/transformers/modeling_tf_albert.py src/transformers/modeling_tf_albert.py +8 -8

No files found.
--- a/src/transformers/modeling_tf_albert.py
+++ b/src/transformers/modeling_tf_albert.py
@@ -506,14 +506,14 @@ class TFAlbertMainLayer(tf.keras.layers.Layer):
        raise NotImplementedError

    def call(
-            self,
-            inputs,
-            attention_mask=None,
-            token_type_ids=None,
-            position_ids=None,
-            head_mask=None,
-            inputs_embeds=None,
-            training=False,
+        self,
+        inputs,
+        attention_mask=None,
+        token_type_ids=None,
+        position_ids=None,
+        head_mask=None,
+        inputs_embeds=None,
+        training=False,
    ):
        if isinstance(inputs, (tuple, list)):
            input_ids = inputs[0]