 float or int) — (batch) timesteps


encoder_hidden_states (torch.FloatTensor) — (batch, sequence_length, feature_dim) encoder hidden states


return_dict (bool, optional, defaults to True) —
Whether or not to return a models.unet_2d_condition.UNet2DConditionOutput instead of a plain tuple.


cross_attention_kwargs (dict, optional) —
A kwargs dictionary that if specified is passed along to the AttentionProcessor as defined under
self.processor 