From ab8e5364cf521e1023560cfddf14bec02824bf34 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 7 Jun 2022 10:59:46 +0000 Subject: [PATCH] correct typo Config Mixin --- src/diffusers/models/unet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/unet.py b/src/diffusers/models/unet.py index 6a42fe3e..470d420c 100644 --- a/src/diffusers/models/unet.py +++ b/src/diffusers/models/unet.py @@ -29,7 +29,7 @@ from torchvision import transforms, utils from PIL import Image from tqdm import tqdm -from ..configuration_utils import ConfigMixinMixin +from ..configuration_utils import ConfigMixin from ..modeling_utils import ModelMixin @@ -175,7 +175,7 @@ class AttnBlock(nn.Module): return x + h_ -class UNetModel(ModelMixin, ConfigMixinMixin): +class UNetModel(ModelMixin, ConfigMixin): def __init__( self, ch=128,