teps used when generating samples with a pre-trained model.   device (str or torch.device, optional) —
The device to which the timesteps and diffusion process parameters (alpha, beta, gamma) should be moved
to.    Sets the discrete timesteps used for the diffusion chain (to be run before inference).  step  < source > ( model_output: FloatTensor timestep: torch.int64 sample: LongTensor generator: Optional = None return_dict: bool = True  ) → VQD