gs, 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 
~models.prior_transformer.PriorTransformerOutput or tuple
  
If return_di