tput of the network. This is often run repeatedly
after making the prediction for the previous timestep.

step_pred

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)
→
SdeVeOutput or tuple

Parameters 

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


timestep (int) — current discrete timestep in the diffusio