fore inference.
The sigmas control the weight of the drift and diffusion components of sample update.

set_timesteps

<
source
>
(
num_inference_steps: int
sampling_eps: float = None
device: typing.Union[str, torch.device] = None

)


Parameters 

num_inference_steps (int) —
the number of diffusion steps used when generating samples with a pre-trained model.


sampling_eps (float, optional) —
final timestep value (overrides value given at Sched