Merge branch 'master' into dev

LongxingTan · Jun 13, 2023 · 3f9fb3b · 3f9fb3b
2 parents ed7acf1 + a5de508
commit 3f9fb3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfts/layers/attention_layer.py b/tfts/layers/attention_layer.py
@@ -15,7 +15,7 @@
 class FullAttention(tf.keras.layers.Layer):
     """Multi-head attention layer"""
 
-    def __init__(self, hidden_size: int, num_heads: int, attention_dropout: float = 0.0, **kwargs):
+    def __init__(self, hidden_size: int, num_heads: int, attention_dropout: float = 0.0):
         """Initialize the layer.
 
         Parameters: