nodes/HowToSD-ComfyUI-Pt-Wrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed the docstring for MHA custom class

Browse Source
This commit is contained in:
HowToSD
2025-04-07 19:37:13 -07:00
parent ec9339a62b
commit 2168836d13
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/pytorch_wrapper/ptn_multihead_attention_custom.py
+1 -1
View File
@@ -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.