Skip to content

vllm_omni.model_executor.models.common.ming.fm

Solver

func instance-attribute

func = func

sigma instance-attribute

sigma = sigma

temperature instance-attribute

temperature = temperature

y0 instance-attribute

y0 = y0

integrate

integrate(t)

apply_sway_sampling

apply_sway_sampling(t, sway_sampling_coef)

build_timesteps

build_timesteps(
    steps,
    device,
    dtype,
    use_epss=True,
    sway_sampling_coef=-1.0,
)

integrate_cfm_steps

integrate_cfm_steps(fn, y0, t, sde_args, sde_rnd, steps)