This commit is contained in:
Alex "mcmonkey" Goodwin 2024-09-16 13:32:54 +09:00 committed by GitHub
parent 5e83837c01
commit cb97d53be2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,
)