  )   Parameters   num_inference_steps (int) —
The number of diffusion steps used when generating samples with a pre-trained model.   device (str or torch.device, optional) —
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: FloatTensor timestep: Union sample: FloatTensor order: in