mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-12 12:37:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove omegaconf dependency and some ci changes.

Browse Source
This commit is contained in:
comfyanonymous
2023-03-13 14:49:18 -04:00
parent 80665081e0
commit 54dbfaf2ec
8 changed files with 26 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/cldm/cldm.py
View File

@@ -59,9 +59,9 @@ class ControlNet(nn.Module):
if context_dim is not None:
assert use_spatial_transformer, 'Fool!! You forgot to use the spatial transformer for your cross-attention conditioning...'
from omegaconf.listconfig import ListConfig
if type(context_dim) == ListConfig:
context_dim = list(context_dim)
# from omegaconf.listconfig import ListConfig
# if type(context_dim) == ListConfig:
# context_dim = list(context_dim)
if num_heads_upsample == -1:
num_heads_upsample = num_heads