Diffusion Pipeline Special


source

DiffusionPipeline_attnPadded

 DiffusionPipeline_attnPadded
                               (scheduler:genQC.scheduler.scheduler.Schedu
                               ler, 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 special DiffusionPipeline with attention masking.


source

DiffusionPipeline_Compilation

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

Back to top