From 7e5db0866ff39f06193da77bc7aa0814db89064b Mon Sep 17 00:00:00 2001 From: David <48716362+oldskool978@users.noreply.github.com> Date: Wed, 5 Nov 2025 22:04:36 -0500 Subject: [PATCH] Update cldm.py cleanup --- comfy/cldm/cldm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/cldm/cldm.py b/comfy/cldm/cldm.py index ec01665e2..a5dc4090a 100644 --- a/comfy/cldm/cldm.py +++ b/comfy/cldm/cldm.py @@ -99,10 +99,10 @@ class ControlNet(nn.Module): super().__init__() assert use_spatial_transformer == True, "use_spatial_transformer has to be true" if use_spatial_transformer: - assert context_dim is not None, 'Fool!! You forgot to include the dimension of your cross-attention conditioning...' + assert context_dim is not None, if context_dim is not None: - assert use_spatial_transformer, 'Fool!! You forgot to use the spatial transformer for your cross-attention conditioning...' + assert use_spatial_transformer, # from omegaconf.listconfig import ListConfig # if type(context_dim) == ListConfig: # context_dim = list(context_dim)