ised
image do not contain a prediction for the masked pixel as the unnoised image cannot be masked.

forward

<
source
>
(
hidden_states: Tensor
encoder_hidden_states: typing.Optional[torch.Tensor] = None
timestep: typing.Optional[torch.LongTensor] = None
class_labels: typing.Optional[torch.LongTensor] = None
cross_attention_kwargs: typing.Dict[str, typing.Any] = None
attention_mask: typing.Optional[torch.Tensor] = None
encoder_attention_mask: 