
The currently predicted image embeddings.   timestep (torch.LongTensor) —
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 is conditioned on.   attention_mask (torch.Bool