From 7a6eb212aedb6a9399f1c6014e6f0f2837bd4da2 Mon Sep 17 00:00:00 2001 From: Ikko Ashimine Date: Wed, 30 Nov 2022 12:23:19 +0900 Subject: [PATCH] fix typo in unet.py ouput -> output --- guided_diffusion/unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/guided_diffusion/unet.py b/guided_diffusion/unet.py index 96b46930..60ccf78c 100644 --- a/guided_diffusion/unet.py +++ b/guided_diffusion/unet.py @@ -327,7 +327,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):