support sdxl controlnet union

2026-03-20 23:58:12 +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/sd_controlnet.py
+++ b/diffsynth/models/sd_controlnet.py
@@ -97,6 +97,7 @@ class SDControlNet(torch.nn.Module):
        self,
        sample, timestep, encoder_hidden_states, conditioning,
        tiled=False, tile_size=64, tile_stride=32,
+        **kwargs
    ):
        # 1. time
        time_emb = self.time_proj(timestep).to(sample.dtype)