def ddim_elem

mlbend-generative@0.1.1.0 · main.bend:369

import mlbend-generative@0.1.1.0/main.bend as MlbendGenerative
import 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)

Uses (5)

Used by (1)