mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-11-12 02:26:46 +08:00
fix typo
This commit is contained in:
parent
5e83837c01
commit
cb97d53be2
@ -858,7 +858,7 @@ class MMDiT(nn.Module):
|
|||||||
context = torch.cat(
|
context = torch.cat(
|
||||||
(
|
(
|
||||||
repeat(self.register, "1 ... -> b ...", b=x.shape[0]),
|
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,
|
1,
|
||||||
)
|
)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user