

One step for the UniC (B(h) version).

multistep_uni_p_bh_update

<
source
>
(
model_output: FloatTensor
prev_timestep: int
sample: FloatTensor
order: int

)
→
torch.FloatTensor

Parameters 

model_output (torch.FloatTensor) —
direct outputs from learned diffusion model at the current timestep.


prev_timestep (int) — previous discrete timestep in the diffusion chain.


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