rating samples with a pre-trained model.


sampling_eps (float, optional) — final timestep value (overrides value given at Scheduler instantiation).



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

step_correct

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

)
→
SdeVeOutput or tuple

Paramete