Diffusion Pipeline

---

source

DiffusionPipeline

 DiffusionPipeline (scheduler:genQC.scheduler.scheduler.Scheduler,
                    model:torch.nn.modules.module.Module,
                    text_encoder:torch.nn.modules.module.Module,
                    embedder:torch.nn.modules.module.Module,
                    device:torch.device, enable_guidance_train=True,
                    guidance_train_p=0.1, cached_text_enc=True)

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

	Type	Default	Details
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