Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

rename to UNetLDMModel

This commit is contained in:
patil-suraj 2022-06-08 11:29:28 +02:00
parent 4ea4429d1a
commit 2f24ce1ce3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/unet_ldm.py
View File

@ -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.