Update unet_2d_blocks.py

This commit is contained in:
kijai
2024-04-07 15:46:25 +03:00
parent f36faeae1e
commit 82ad561ba2
+1 -1
View File
@@ -25,7 +25,7 @@ from diffusers.utils import is_torch_version, logging
from diffusers.utils.torch_utils import apply_freeu from diffusers.utils.torch_utils import apply_freeu
from diffusers.models.activations import get_activation from diffusers.models.activations import get_activation
from diffusers.models.attention_processor import Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0 from diffusers.models.attention_processor import Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
from diffusers.models.dual_transformer_2d import DualTransformer2DModel from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
from diffusers.models.normalization import AdaGroupNorm from diffusers.models.normalization import AdaGroupNorm
from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
from .transformer_2d import Transformer2DModel from .transformer_2d import Transformer2DModel