, num classes), optional) —
Optional class labels to be applied as an embedding in AdaLayerZeroNorm. Used to indicate class labels
conditioning.


return_dict (bool, optional, defaults to True) —
Whether or not to return a models.unet_2d_condition.UNet2DConditionOutput instead of a plain tuple.


Returns

~models.transformer_2d.TransformerTemporalModelOutput or tuple



~models.transformer_2d.TransformerTemporalModelOutput if return_dict is Tru