From 2168836d13b7a58cd3d7ebf0b33c99c7ceb5f949 Mon Sep 17 00:00:00 2001
From: HowToSD <support@candee.ai>
Date: Mon, 7 Apr 2025 19:37:13 -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 e410ca0..f27620b 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.
+    Implements Multihead attention.
 
     Attributes:
         embed_dim (int): Hidden dimension (d_model) for each token of the Transformer.