Compilation Diffusion Pipeline
Special extension to
DiffusionPipeline
.
Diffusion Pipeline - Compilation
DiffusionPipeline_Compilation
DiffusionPipeline_Compilation (scheduler:genQC.scheduler.scheduler.Sched uler, 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 special DiffusionPipeline
that accounts for unitary conditions, i.e. compilation.
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 |