Remove some useless imports

This commit is contained in:
comfyanonymous 2023-02-24 12:36:55 -05:00
parent 87b00b37f6
commit cf5a211efc

View File

@ -1,7 +1,6 @@
#taken from: https://github.com/lllyasviel/ControlNet #taken from: https://github.com/lllyasviel/ControlNet
#and modified #and modified
import einops
import torch import torch
import torch as th import torch as th
import torch.nn as nn import torch.nn as nn
@ -13,8 +12,6 @@ from ldm.modules.diffusionmodules.util import (
timestep_embedding, timestep_embedding,
) )
from einops import rearrange, repeat
from torchvision.utils import make_grid
from ldm.modules.attention import SpatialTransformer from ldm.modules.attention import SpatialTransformer
from ldm.modules.diffusionmodules.openaimodel import UNetModel, TimestepEmbedSequential, ResBlock, Downsample, AttentionBlock from ldm.modules.diffusionmodules.openaimodel import UNetModel, TimestepEmbedSequential, ResBlock, Downsample, AttentionBlock
from ldm.models.diffusion.ddpm import LatentDiffusion from ldm.models.diffusion.ddpm import LatentDiffusion