rename to UNetLDMModel

2022-06-08 11:29:28 +02:00 · 2022-06-08 11:29:28 +02:00 · 2f24ce1ce3
parent 4ea4429d1a
commit 2f24ce1ce3
1 changed files with 1 additions and 1 deletions
--- 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.