s attention features.   attention_head_dim (int, optional, defaults to 64) — The dimension of the attention heads.   num_attention_heads (int, optional) — The number of attention heads.    A conditional 3D UNet model that takes a noisy sample, conditional state, and a timestep and returns a sample
shaped output. This model inherits from ModelMixin. Check the superclass documentation for it’s generic methods implemented
for all models (such as d