From ec9339a62be9d98d3500ec27cd0d6af7f6276d56 Mon Sep 17 00:00:00 2001
From: HowToSD <support@candee.ai>
Date: Mon, 7 Apr 2025 19:36:30 -0700
Subject: [PATCH] Fixed the docstring for MHA custom class

---
 modules/pytorch_wrapper/ptn_multihead_attention_custom.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/pytorch_wrapper/ptn_multihead_attention_custom.py b/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
index 7b4b86c..e410ca0 100644
--- a/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
+++ b/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
@@ -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.