j_embedding: FloatTensor
encoder_hidden_states: FloatTensor
attention_mask: typing.Optional[torch.BoolTensor] = None
return_dict: bool = True

)
→
PriorTransformerOutput or tuple

Parameters 

hidden_states (torch.FloatTensor of shape (batch_size, embedding_dim)) —
x_t, the currently predicted image embeddings.


timestep (torch.long) —
Current denoising step.


proj_embedding (torch.FloatTensor of shape (batch_size, embedding_dim)) —
Projected