FloatTensor of shape (batch_size, num_embeddings, embedding_dim)) —
Hidden states of the text embeddings the denoising process is conditioned on.


attention_mask (torch.BoolTensor of shape (batch_size, num_embeddings)) —
Text mask for the text embeddings.


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


Returns

PriorTransformerOutput or tu