support sdxl controlnet union

2026-03-23 00:58:11 +00:00 · 2024-08-01 10:01:39 +08:00
parent 60d7bb52d6
commit 6f79fd6d77
10 changed files with 408 additions and 17 deletions
--- a/diffsynth/models/sdxl_unet.py
+++ b/diffsynth/models/sdxl_unet.py
@@ -83,6 +83,8 @@ class SDXLUNet(torch.nn.Module):
        self.conv_act = torch.nn.SiLU()
        self.conv_out = torch.nn.Conv2d(320, 4, kernel_size=3, padding=1)

+        self.is_kolors = is_kolors
+
    def forward(
        self,
        sample, timestep, encoder_hidden_states, add_time_id, add_text_embeds,