gs 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 tuple



PriorTransformerOutput if return_dict is True, otherwise a tuple. When
returning a tuple, the f