 tuple

Parameters 

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


timestep (torch.FloatTensor or 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.UNet3DConditionOutput instead of a plain tuple.


cross_at