fix typo in unet.py

ouput -> output
jS5t3r · Apr 14, 2023 · 5f9fb5a · 5f9fb5a
1 parent 6d26080
commit 5f9fb5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cm/unet.py b/cm/unet.py
@@ -396,7 +396,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):