tion.


norm_eps (float, optional, defaults to 1e-5) — The epsilon for the normalization.


resnet_time_scale_shift (str, optional, defaults to "default") — Time scale shift config
for resnet blocks, see ResnetBlock2D. Choose from default or scale_shift.



UNet2DModel is a 2D UNet model that takes in a noisy sample and a timestep and returns sample shaped output.
This model inherits from ModelMixin. Check the superclass documentation for the g