Update model.py

model.py

@@ -40,8 +40,8 @@ from diffusers.models.embeddings import (
 from diffusers.models.activations import get_activation
 from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor, Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
 from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
-from diffusers.models.transformer_2d import Transformer2DModel
-from diffusers.models.dual_transformer_2d import DualTransformer2DModel
+from diffusers.models.transformers.transformer_2d import Transformer2DModel
+from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
 
 
 class CrossAttnDownBlock2DWithFlow(nn.Module):