def ddim_elem
import mlbend-generative@0.1.1.0/main.bend as MlbendGenerativeimport Base
369# one element of the step: x0 folded into x_prev
370def ddim_elem(+co: Step, +x: F32, +eps: F32, +z: F32) -> F32:
371 match co:
372 case Step{c0, c1, c2, c3, sig}:
373 +x0 = ((x - (c1 * eps : F32)) * c0 : F32)
374 (((x0 * c2 : F32) + (c3 * eps : F32) : F32) + (sig * z : F32) : F32)