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