From 82ad561ba27c9b11a2a832f06a22360f536c3487 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 7 Apr 2024 15:46:25 +0300
Subject: [PATCH] Update unet_2d_blocks.py

---
 models/unet_2d_blocks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/unet_2d_blocks.py b/models/unet_2d_blocks.py
index f658be3..2c29c0f 100644
--- a/models/unet_2d_blocks.py
+++ b/models/unet_2d_blocks.py
@@ -25,7 +25,7 @@ from diffusers.utils import is_torch_version, logging
 from diffusers.utils.torch_utils import apply_freeu
 from diffusers.models.activations import get_activation
 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.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
 from .transformer_2d import Transformer2DModel