From 267a26e61ffcc8429792fab1bb66db10b1bd0cdc Mon Sep 17 00:00:00 2001 From: PAseer <128199180+NBSTpeterhill@users.noreply.github.com> Date: Thu, 3 Oct 2024 17:12:56 +0200 Subject: [PATCH 1/2] Update unet_garm_2d_blocks.py diffusers import changes --- pipelines_ootd/unet_garm_2d_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipelines_ootd/unet_garm_2d_blocks.py b/pipelines_ootd/unet_garm_2d_blocks.py index 06dddd9..974d989 100644 --- a/pipelines_ootd/unet_garm_2d_blocks.py +++ b/pipelines_ootd/unet_garm_2d_blocks.py @@ -26,7 +26,7 @@ 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 diffusers.models.transformer_2d import Transformer2DModel From 84eac3f8218d6504420d38939c5260848ae13bdf Mon Sep 17 00:00:00 2001 From: PAseer <128199180+NBSTpeterhill@users.noreply.github.com> Date: Thu, 3 Oct 2024 17:13:23 +0200 Subject: [PATCH 2/2] Update unet_vton_2d_blocks.py diffusers imports changes --- pipelines_ootd/unet_vton_2d_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipelines_ootd/unet_vton_2d_blocks.py b/pipelines_ootd/unet_vton_2d_blocks.py index 57e2555..0fc7425 100644 --- a/pipelines_ootd/unet_vton_2d_blocks.py +++ b/pipelines_ootd/unet_vton_2d_blocks.py @@ -26,7 +26,7 @@ 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 diffusers.models.transformer_2d import Transformer2DModel