ional, defaults to 1e-5) — The epsilon to use for the normalization.


cross_attention_dim (int, optional, defaults to 1280) — The dimension of the cross attention features.


attention_head_dim (int, optional, defaults to 8) — The dimension of the attention heads.



UNet3DConditionModel is a conditional 2D UNet model that takes in a noisy sample, conditional state, and a timestep
and returns sample shaped output.
This model inherits from Mode