_inference_steps: int
device: typing.Union[str, torch.device] = None
num_train_timesteps: typing.Optional[int] = 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 timesteps used for the diffusion chain. Supporting f