From 2f24ce1ce3988b2e622bc9a39f29255494bc62a8 Mon Sep 17 00:00:00 2001
From: patil-suraj <surajp815@gmail.com>
Date: Wed, 8 Jun 2022 11:29:28 +0200
Subject: [PATCH] rename to UNetLDMModel

---
 src/diffusers/models/unet_ldm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/diffusers/models/unet_ldm.py b/src/diffusers/models/unet_ldm.py
index d6006a49..0b0b3bc1 100644
--- a/src/diffusers/models/unet_ldm.py
+++ b/src/diffusers/models/unet_ldm.py
@@ -720,7 +720,7 @@ class QKVAttentionLegacy(nn.Module):
         return count_flops_attn(model, _x, y)
 
 
-class UNetModel(ModelMixin, ConfigMixin):
+class UNetLDMModel(ModelMixin, ConfigMixin):
     """
     The full UNet model with attention and timestep embedding.
     :param in_channels: channels in the input Tensor.