one
cross_attention_kwargs: typing.Union[typing.Dict[str, typing.Any], NoneType] = None
down_block_additional_residuals: typing.Optional[typing.Tuple[torch.Tensor]] = None
mid_block_additional_residual: typing.Optional[torch.Tensor] = None
return_dict: bool = True

)
→
~models.unet_2d_condition.UNet3DConditionOutput or tuple

Parameters 

sample (torch.FloatTensor) — (batch, num_frames, channel, height, width) noisy inputs tensor


timestep (to