Diffusion Pipeline

---

source

DiffusionPipeline

def DiffusionPipeline(
    scheduler:Scheduler, model:Module, text_encoder:Module,
    embedder:Module, # clr embeddings or a VAE for latent diffusion
    device:device, enable_guidance_train:bool=True, guidance_train_p:float=0.1, cached_text_enc:bool=True
):

A Pipeline for diffusion models. Implements train and inference functions. Diffusion parameters are defined inside a Scheduler object.