Patch ALBERT with heads in TensorFlow

2020-02-19 18:46:25 -05:00
parent 1abd53b1aa
commit 2708b44ee9
1 changed files with 8 additions and 8 deletions
@@ -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]