From c2f0278dff5d3d7c8dfe6f36d7d53b448dd69857 Mon Sep 17 00:00:00 2001
From: Yalong Song <144522198+SolveProb@users.noreply.github.com>
Date: Tue, 6 Aug 2024 14:34:42 +0800
Subject: [PATCH] [bugfix] No module named 'diffusers.models.unet_2d_condition'

The latest version of diffusers has modified its file organization structure.
---
 mmagic/models/editors/vico/vico_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmagic/models/editors/vico/vico_utils.py b/mmagic/models/editors/vico/vico_utils.py
index bc0fec65cd..387b76dfc8 100644
--- a/mmagic/models/editors/vico/vico_utils.py
+++ b/mmagic/models/editors/vico/vico_utils.py
@@ -7,7 +7,7 @@
 import torch.nn.functional as F
 from diffusers import Transformer2DModel
 from diffusers.models.attention import Attention, BasicTransformerBlock
-from diffusers.models.unet_2d_condition import UNet2DConditionOutput
+from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput
 from diffusers.utils import BaseOutput, is_torch_version