~schedulers.scheduling_utils.VQDiffusionSchedulerOutput or tuple

Parameters 

t (torch.long) —
The timestep that determines which transition matrices are used.
x_t — (torch.LongTensor of shape (batch size, num latent pixels)):
The classes of each latent pixel at time t
generator — (torch.Generator or None):
RNG for the noise applied to p(x_{t-1} | x_t) before it is sampled from.


return_dict (bool) —
option for returning tuple rather than VQD