wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-13 23:12:35 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update cldm.py

Browse Source 
cleanup
This commit is contained in:
David 2025-11-05 22:04:36 -05:00 committed by GitHub
parent 09dc24c8a9
commit 7e5db0866f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/cldm/cldm.py
View File

@ -99,10 +99,10 @@ class ControlNet(nn.Module):
super().__init__() super().__init__()
assert use_spatial_transformer == True, "use_spatial_transformer has to be true" assert use_spatial_transformer == True, "use_spatial_transformer has to be true"
if use_spatial_transformer: 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: 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 # from omegaconf.listconfig import ListConfig
# if type(context_dim) == ListConfig: # if type(context_dim) == ListConfig:
# context_dim = list(context_dim) # context_dim = list(context_dim)