—
The device to which the timesteps should be moved to. If None, the timesteps are not moved.    Sets the discrete timesteps used for the diffusion chain (to be run before inference).  step  < source > ( model_output: Union timestep: Union sample: Union return_dict: bool = True  ) → SchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from learned diffusion model.   timestep (float) —
The current discrete 