 inference.  set_timesteps  < source > ( num_inference_steps: int device: Union = None num_train_timesteps: Optional = None  )   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 chai