rd  < source > ( hidden_states timestep: Union proj_embedding: FloatTensor encoder_hidden_states: Optional = None attention_mask: Optional = None return_dict: bool = True  ) → ~models.prior_transformer.PriorTransformerOutput or tuple  Parameters   hidden_states (torch.FloatTensor of shape (batch_size, embedding_dim)) —
The currently predicted image embeddings.   timestep (torch.LongTensor) —
Current denoising step.   proj_embedding (torch.Float