Skip to content

GitLab

Commit 2f24ce1c authored by patil-suraj's avatar patil-suraj
Browse files

rename to UNetLDMModel

parent 4ea4429d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/unet_ldm.py
View file @ 2f24ce1c
...@@ -720,7 +720,7 @@ class QKVAttentionLegacy(nn.Module): ...@@ -720,7 +720,7 @@ class QKVAttentionLegacy(nn.Module):
return count_flops_attn(model, _x, y) return count_flops_attn(model, _x, y)
class UNetModel(ModelMixin, ConfigMixin): class UNetLDMModel(ModelMixin, ConfigMixin):
""" """
The full UNet model with attention and timestep embedding. The full UNet model with attention and timestep embedding.
:param in_channels: channels in the input Tensor. :param in_channels: channels in the input Tensor.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment