Skip to content

GitLab

Commit ade7aa1b authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove useless import.

parent faa57430
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
comfy/cldm/cldm.py
View file @ ade7aa1b
...@@ -13,7 +13,6 @@ from ..ldm.modules.diffusionmodules.util import ( ...@@ -13,7 +13,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 ..ldm.util import exists
from ..ldm.cascade.common import OptimizedAttention
from collections import OrderedDict from collections import OrderedDict
import comfy.ops import comfy.ops
from comfy.ldm.modules.attention import optimized_attention from comfy.ldm.modules.attention import optimized_attention
...@@ -388,7 +387,7 @@ class ControlNet(nn.Module): ...@@ -388,7 +387,7 @@ class ControlNet(nn.Module):
emb = self.time_embed(t_emb) emb = self.time_embed(t_emb)
guided_hint = None 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", []) control_type = kwargs.get("control_type", [])
emb += self.control_add_embedding(control_type, emb.dtype, emb.device) emb += self.control_add_embedding(control_type, emb.dtype, emb.device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment