ddings.
timestep_cond — (torch.Tensor, optional, defaults to None):
Conditional embeddings for timestep. If provided, the embeddings will be summed with the samples passed
through the self.time_embedding layer to obtain the timestep embeddings.   attention_mask (torch.Tensor, optional, defaults to None) —
An attention mask of shape (batch, key_tokens) is applied to encoder_hidden_states. If 1 the mask
is kept, otherwise if 0 it is discarded. Ma