tTensor 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 embedding vector the denoising process is conditioned on.


encoder_hidden_states (torch.FloatTensor of shape (batch_size, num_embeddings, embedding_dim)) —
Hidden states of the text embeddings the denoising process i