r not to return a models.unet_2d_condition.UNet2DConditionOutput instead of a plain tuple.


Returns

Transformer2DModelOutput or tuple



Transformer2DModelOutput if return_dict is True, otherwise a tuple. When
returning a tuple, the first element is the sample tensor.



Transformer2DModelOutput
	

class diffusers.models.transformer_2d.Transformer2DModelOutput

<
source
>
(
sample: FloatTensor

)


Parameters 

sample (torch.FloatTensor of sh