298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove useless code.

Browse Source
This commit is contained in:
comfyanonymous 2024-07-04 15:14:13 -04:00
parent 24b969d3da
commit 739b76630e
1 changed files with 0 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/cldm/mmdit.py
View File

@ -56,15 +56,6 @@ class ControlNet(comfy.ldm.modules.diffusionmodules.mmdit.MMDiT):
if context is not None:
context = self.context_embedder(context)
if self.register_length > 0:
context = torch.cat(
(
repeat(self.register, "1 ... -> b ...", b=x.shape[0]),
default(context, torch.Tensor([]).type_as(x)),
),
1,
)
output = []
blocks = len(self.joint_blocks)