correct typo Config Mixin

ab8e5364 · Patrick von Platen · 5b1af9ab · ab8e5364
Commit ab8e5364 authored Jun 07, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/diffusers/models/unet.py src/diffusers/models/unet.py +2 -2

No files found.
--- 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,