)


Parameters 

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



Sets the discrete timesteps used for the diffusion chain. Supporting function to be run before inference.

step

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor
eta: float = 0.0
use_clipped_model_output: bool = False
generator = None
variance_noise: typing.Optional[torch.FloatTensor] = Non