timesteps are moved to.


custom_timesteps (List[int], optional) —
custom timesteps used to support arbitrary spacing between timesteps. If None, then the default
timestep spacing strategy of equal spacing between timesteps is used. If passed, num_inference_steps
must be None.



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

step

<
source
>
(
model_output: FloatTensor
timestep: int
s