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  ) → VQDiffusionSchedulerOutput or tuple  Parameters   t (torch.long) —