modelzoo.vision.pytorch.dit.pipeline

Classes

DiffusionPipeline

param sampler Instance of one of the supported samplers.

previous

modelzoo.vision.pytorch.dit.layers.vae.utils.DiagonalGaussianDistribution

next

modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline