Update cldm/cldm.py

cldm/cldm.py

@@ -336,6 +336,8 @@ class ControlNet(nn.Module):
         for module, zero_conv in zip(self.input_blocks, self.zero_convs):
             if guided_hint is not None:
                 h = module(h, emb, context)
+                print("h shape:", h.shape)
+                print("guided hint shape:", guided_hint.shape)
                 h += guided_hint
                 guided_hint = None
             else: