jcm/models/utils.py [178:186]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    sde, model, params, states, train=False, return_state=False, pred_t=None
):
    assert isinstance(
        sde, sde_lib.KVESDE
    ), "Only KVE SDE is supported for building the denoiser"
    model_fn = get_model_fn(model, params, states, train=train)

    if pred_t is None:
        pred_t = sde.t_min
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



jcm/models/utils.py [210:218]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    sde, model, params, states, train=False, return_state=False, pred_t=None
):
    assert isinstance(
        sde, sde_lib.KVESDE
    ), "Only KVE SDE is supported for building the denoiser"
    model_fn = get_model_fn(model, params, states, train=train)

    if pred_t is None:
        pred_t = sde.t_min
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -