DiffusionIterator#
- class deepinv.sampling.DiffusionIterator(cur_params=None, clip=None)[source]#
Bases:
SamplingIterator
Helper class used by
deepinv.sampling.DiffusionSampler
to interface diffusion models with thedeepinv.sampling.BaseSampling
framework.Note
Users should typically interact with
deepinv.sampling.DiffusionSampler
rather than this class directly.