Update models/unet_2d_blocks.py

models/unet_2d_blocks.py

@@ -20,11 +20,11 @@ from torch import nn
 
 from diffusers.utils import is_torch_version, logging
 from diffusers.models.activations import get_activation
-from diffusers.models.attention import AdaGroupNorm
+from diffusers.models.normalization import AdaGroupNorm
 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.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
-from diffusers.models.transformer_2d import Transformer2DModel
+from diffusers.models.transformers.transformer_2d import Transformer2DModel
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name