rect output from learned diffusion model.


timestep (int) — current 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 (DPM-Solver / DPM-Solver++) needs.
DPM-Solver is designed to discretize an integral of the noise prediction mod