iffusion chain. Supporting function to be run before inference.

step

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor
generator = None
return_dict: bool = True
**kwargs

)
→
~schedulers.scheduling_utils.DDPMSchedulerOutput or tuple

Parameters 

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


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


sample (torch.FloatTensor)