Diffusion Pipeline

---

source

DiffusionPipeline

 DiffusionPipeline (scheduler:genQC.scheduler.scheduler.Scheduler,
                    model:torch.nn.modules.module.Module,
                    text_encoder: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.