From 739b76630e0298561c39d0d3341bc2856e55d995 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 4 Jul 2024 15:14:13 -0400
Subject: [PATCH] Remove useless code.

---
 comfy/cldm/mmdit.py | 9 ---------
 1 file changed, 9 deletions(-)

diff --git a/comfy/cldm/mmdit.py b/comfy/cldm/mmdit.py
index ee0282b..025c2fb 100644
--- a/comfy/cldm/mmdit.py
+++ b/comfy/cldm/mmdit.py
@@ -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)