modelzoo.vision.pytorch.dit.pipeline

Classes

DiffusionPipeline

param sampler Instance of one of the supported samplers.

previous

modelzoo.vision.pytorch.dit.modeling_dit.DiT

next

modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline