correct naming in glide

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 ConfigMixin
+from ..configuration_utils import ConfigMixinMixin
 from ..modeling_utils import ModelMixin
 
 
@@ -175,7 +175,7 @@ class AttnBlock(nn.Module):
         return x + h_
 
 
-class UNetModel(ModelMixin, ConfigMixin):
+class UNetModel(ModelMixin, ConfigMixinMixin):
     def __init__(
         self,
         ch=128,

src/diffusers/models/unet_glide.py

@@ -5,8 +5,8 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
-from ..configuration_utils import Config
-from ..modeling_utils import PreTrainedModel
+from ..configuration_utils import ConfigMixin
+from ..modeling_utils import ModelMixin
 
 
 def convert_module_to_f16(l):
@@ -388,7 +388,7 @@ class QKVAttention(nn.Module):
         return a.reshape(bs, -1, length)
 
 
-class UNetGLIDEModel(PreTrainedModel, Config):
+class UNetGLIDEModel(ModelMixin, ConfigMixin):
     """
     The full UNet model with attention and timestep embedding.