 to which the timesteps should be moved to. If None, the timesteps are not moved.



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

step

<
source
>
(
model_output: typing.Union[torch.FloatTensor, numpy.ndarray]
timestep: typing.Union[float, torch.FloatTensor]
sample: typing.Union[torch.FloatTensor, numpy.ndarray]
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)