orch.device] = None
timesteps: typing.Optional[typing.List[int]] = None

)


Parameters 

num_inference_steps (Optional[int]) —
the number of diffusion steps used when generating samples with a pre-trained model. If passed, then
timesteps must be None.


device (str or torch.device, optional) —
the device to which the timesteps are moved to.


custom_timesteps (List[int], optional) —
custom timesteps used to support arbitrary spacing between ti