in jcm/sampling.py [0:0]
def __init__(self, sde, score_fn, snr, n_steps):
super().__init__(sde, score_fn, snr, n_steps)
if (
not isinstance(sde, sde_lib.VPSDE)
and not isinstance(sde, sde_lib.VESDE)
and not isinstance(sde, sde_lib.subVPSDE)
):
raise NotImplementedError(
f"SDE class {sde.__class__.__name__} not yet supported."
)