he timesteps are not moved.



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

step

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor
return_dict: bool = True

)
→
~scheduling_utils.SchedulerOutput or tuple

Parameters 

model_output (torch.FloatTensor) — direct output from learned diffusion model.


timestep (int) — current discrete timestep in the diffusion chain.


sam