qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove useless import.

This commit is contained in:
comfyanonymous 2024-07-09 11:05:05 -04:00
parent faa57430b0
commit ade7aa1b0c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/cldm/cldm.py
View File

@ -13,7 +13,6 @@ from ..ldm.modules.diffusionmodules.util import (
from ..ldm.modules.attention import SpatialTransformer
from ..ldm.modules.diffusionmodules.openaimodel import UNetModel, TimestepEmbedSequential, ResBlock, Downsample
from ..ldm.util import exists
from ..ldm.cascade.common import OptimizedAttention
from collections import OrderedDict
import comfy.ops
from comfy.ldm.modules.attention import optimized_attention
@ -388,7 +387,7 @@ class ControlNet(nn.Module):
emb = self.time_embed(t_emb)
guided_hint = None
if self.control_add_embedding is not None:
if self.control_add_embedding is not None: #Union Controlnet
control_type = kwargs.get("control_type", [])
emb += self.control_add_embedding(control_type, emb.dtype, emb.device)