Fixed the docstring for MHA custom class

2025-04-07 19:36:30 -07:00
parent 080a23778a
commit ec9339a62b
1 changed files with 1 additions and 1 deletions
@@ -6,7 +6,7 @@ from einops import einops

 class PtnMultiheadAttentionCustomDef(nn.Module):
    """
-    Implements MultiheadAttention wrapper.
+    Implements MultiheadAttention.

    Attributes:
        embed_dim (int): Hidden dimension (d_model) for each token of the Transformer.