From 0e86809cb6600a16ca22fc905417e28632fc74fb Mon Sep 17 00:00:00 2001 From: ClownsharkBatwing Date: Thu, 3 Jul 2025 18:39:57 -0400 Subject: [PATCH] Add files via upload --- style_transfer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/style_transfer.py b/style_transfer.py index 564bdb7..d69ff01 100644 --- a/style_transfer.py +++ b/style_transfer.py @@ -1782,8 +1782,8 @@ class StyleMMDiT_Model(Stylizer): y0_style_embed = self.Retrojector.embed(y0_style) denoised_embed = self.Retrojector.embed(denoised) - - embed = torch.cat([denoised_embed, y0_style_embed], dim=0) + B,HW,C = y0_style_embed.shape + embed = torch.cat([denoised_embed, y0_style_embed.view(1,B*HW,C)[:,::B,:]], dim=0) method = getattr(self, mode) embed = method(embed) return self.Retrojector.unembed(embed[0:1])