mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 13:05:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Controlnet refactor.

Browse Source
This commit is contained in:
comfyanonymous
2024-06-25 17:02:05 -04:00
parent 97b409cd48
commit 66aaa14001
4 changed files with 24 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/cascade/controlnet.py
View File

@@ -90,4 +90,4 @@ class ControlNet(nn.Module):
proj_outputs = [None for _ in range(max(self.proj_blocks) + 1)]
for i, idx in enumerate(self.proj_blocks):
proj_outputs[idx] = self.projections[i](x)
return proj_outputs
return {"input": proj_outputs[::-1]}