github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 01:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix typo s/keywrods/keywords etc. (#5429)

Browse Source
This commit is contained in:
digger yu
2024-03-12 11:25:16 +08:00
committed by GitHub
parent da885ed540
commit 385e85afd4
17 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/diffusion/ldm/modules/diffusionmodules/openaimodel.py
View File

@@ -338,7 +338,7 @@ def count_flops_attn(model, _x, y):
class QKVAttentionLegacy(nn.Module):
"""
A module which performs QKV attention. Matches legacy QKVAttention + input/ouput heads shaping
A module which performs QKV attention. Matches legacy QKVAttention + input/output heads shaping
"""
def __init__(self, n_heads):