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.