xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 02:00:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Missed one occurance of "exists"

Browse Source 
Missed one occurance of "exists"
This commit is contained in:
John Slegers 2024-08-13 14:46:32 +02:00
parent 0f874d99a6
commit dd156556ca
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/cldm/cldm.py
View File

@ -12,7 +12,6 @@ from ..ldm.modules.diffusionmodules.util import (
from ..ldm.modules.attention import SpatialTransformer from ..ldm.modules.attention import SpatialTransformer
from ..ldm.modules.diffusionmodules.openaimodel import UNetModel, TimestepEmbedSequential, ResBlock, Downsample from ..ldm.modules.diffusionmodules.openaimodel import UNetModel, TimestepEmbedSequential, ResBlock, Downsample
from ..ldm.util import exists
from .control_types import UNION_CONTROLNET_TYPES from .control_types import UNION_CONTROLNET_TYPES
from collections import OrderedDict from collections import OrderedDict
import comfy.ops import comfy.ops
@ -234,12 +233,12 @@ class ControlNet(nn.Module):
if legacy: if legacy:
#num_heads = 1 #num_heads = 1
dim_head = ch // num_heads if use_spatial_transformer else num_head_channels dim_head = ch // num_heads if use_spatial_transformer else num_head_channels
if exists(disable_self_attentions): if disable_self_attentions is not None:
disabled_sa = disable_self_attentions[level] disabled_sa = disable_self_attentions[level]
else: else:
disabled_sa = False disabled_sa = False
if not exists(num_attention_blocks) or nr < num_attention_blocks[level]: if num_attention_blocks is not None or nr < num_attention_blocks[level]:
layers.append( layers.append(
SpatialTransformer( SpatialTransformer(
ch, num_heads, dim_head, depth=num_transformers, context_dim=context_dim, ch, num_heads, dim_head, depth=num_transformers, context_dim=context_dim,
@ -434,4 +433,3 @@ class ControlNet(nn.Module):
out_middle.append(self.middle_block_out(h, emb, context)) out_middle.append(self.middle_block_out(h, emb, context))
return {"middle": out_middle, "output": out_output} return {"middle": out_middle, "output": out_output}