From cb97d53be2182d0427d3fd921c20f9dcba409bd3 Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" <4000772+mcmonkey4eva@users.noreply.github.com> Date: Mon, 16 Sep 2024 13:32:54 +0900 Subject: [PATCH] fix typo --- comfy/ldm/modules/diffusionmodules/mmdit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/modules/diffusionmodules/mmdit.py b/comfy/ldm/modules/diffusionmodules/mmdit.py index 94a38726e..aecf1ec03 100644 --- a/comfy/ldm/modules/diffusionmodules/mmdit.py +++ b/comfy/ldm/modules/diffusionmodules/mmdit.py @@ -858,7 +858,7 @@ class MMDiT(nn.Module): context = torch.cat( ( repeat(self.register, "1 ... -> b ...", b=x.shape[0]), - torch.Tensor([]).type_as(x) is context is None else context, + torch.Tensor([]).type_as(x) if context is None else context, ), 1, )