num_attention_heads (int, optional):
The number of attention heads. If not defined, defaults to attention_head_dim
resnet_time_scale_shift (str, optional, defaults to "default"): Time scale shift config
for ResNet blocks (see ResnetBlock2D). Choose from default or scale_shift.
class_embed_type (str, optional, defaults to None):
The type of class embedding to use which is ultimately summed with the time embeddings. Choose from None,
"timestep", 