nt discrete timestep in the diffusion chain.


sample (torch.FloatTensor) —
current instance of sample being created by diffusion process.


Returns

torch.FloatTensor



the converted model output.


Convert the model output to the corresponding type that the algorithm PC needs.

multistep_uni_c_bh_update

<
source
>
(
this_model_output: FloatTensor
this_timestep: int
last_sample: FloatTensor
this_sample: FloatTensor
order: int

)
→
torch.Floa