From 914942feb3db68f9602b852d1c2b717cb3da3802 Mon Sep 17 00:00:00 2001 From: anton-l Date: Wed, 23 Nov 2022 12:21:01 +0100 Subject: [PATCH] DualTransformer(nn.Module) --- src/diffusers/models/unet_2d_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/unet_2d_blocks.py b/src/diffusers/models/unet_2d_blocks.py index c645f9f607..0988cbb0ab 100644 --- a/src/diffusers/models/unet_2d_blocks.py +++ b/src/diffusers/models/unet_2d_blocks.py @@ -15,7 +15,7 @@ import numpy as np import torch from torch import nn -from .attention import AttentionBlock, Transformer2DModel +from .attention import AttentionBlock, Transformer2DModel, DualTransformer2DModel from .resnet import Downsample2D, FirDownsample2D, FirUpsample2D, ResnetBlock2D, Upsample2D