Index

A

AblatedDenoiser (class in azula.plugins.adm) ABSampler (class in azula.sample) adams_bashforth() (azula.sample.ABSampler static method) AutoEncoder (class in azula.plugins.eldm) (class in azula.plugins.flux) (class in azula.plugins.sana) (class in azula.plugins.sd) azula module azula.debug module azula.denoise module azula.guidance module azula.guidance.cfg module azula.guidance.diffpir module azula.guidance.dps module azula.guidance.mmps module azula.guidance.pgdm module azula.guidance.repaint module azula.guidance.tds module azula.guidance.tmpd module azula.hub module azula.linalg module

azula.linalg.solve module azula.nn module azula.nn.attention module azula.nn.embedding module azula.nn.layers module azula.nn.unet module azula.nn.utils module azula.noise module azula.plugins module azula.plugins.adm module azula.plugins.edm module azula.plugins.eldm module azula.plugins.flux module azula.plugins.sana module azula.plugins.sd module azula.plugins.utils module azula.plugins.vdm module azula.sample module

C

CFGDenoiser (class in azula.guidance.cfg) cg() (in module azula.linalg.solve)

ConvNd() (in module azula.nn.layers) CosineSchedule (class in azula.noise) cpu_offload() (in module azula.nn.utils)

D

DDIMSampler (class in azula.sample) DDPMSampler (class in azula.sample) DecaySchedule (class in azula.noise) decode() (azula.plugins.eldm.AutoEncoder method) (azula.plugins.flux.AutoEncoder method) (azula.plugins.sana.AutoEncoder method) (azula.plugins.sd.AutoEncoder method)

DiffPIRDenoiser (class in azula.guidance.diffpir) download() (in module azula.hub) DPSSampler (class in azula.guidance.dps)

E

EABSampler (class in azula.sample) ElucidatedDenoiser (class in azula.plugins.edm) ElucidatedLatentDenoiser (class in azula.plugins.eldm) ElucidatedSchedule (class in azula.plugins.edm) encode() (azula.plugins.eldm.AutoEncoder method) (azula.plugins.flux.AutoEncoder method) (azula.plugins.sana.AutoEncoder method) (azula.plugins.sd.AutoEncoder method)

EulerSampler (class in azula.sample) exponential_adams_bashforth() (azula.sample.EABSampler static method)

F

FluxDenoiser (class in azula.plugins.flux) forward() (azula.denoise.GaussianDenoiser method) (azula.denoise.PreconditionedDenoiser method) (azula.guidance.cfg.CFGDenoiser method) (azula.guidance.tds.TDSSampler method) (azula.nn.attention.MultiheadSelfAttention method) (azula.nn.embedding.SineEncoding method) (azula.nn.layers.LayerNorm method) (azula.nn.layers.RMSNorm method) (azula.nn.unet.UNet method) (azula.nn.unet.UNetBlock method) (azula.noise.Schedule method) (azula.plugins.adm.AblatedDenoiser method) (azula.plugins.edm.ElucidatedDenoiser method) (azula.plugins.eldm.ElucidatedLatentDenoiser method) (azula.plugins.flux.FluxDenoiser method) (azula.plugins.flux.TextEncoder method) (azula.plugins.sana.SanaDenoiser method) (azula.plugins.sana.TextEncoder method) (azula.plugins.sd.StableDenoiser method) (azula.plugins.sd.TextEncoder method) (azula.sample.Sampler method)

G

Gaussian (class in azula.denoise) GaussianDenoiser (class in azula.denoise)

get_hub_dir() (in module azula.hub) gmres() (in module azula.linalg.solve)

H

HeunSampler (class in azula.sample)

I

init() (azula.sample.Sampler method)

L

LayerNorm (class in azula.nn.layers) load_cards() (in module azula.plugins.utils) load_model() (in module azula.plugins.adm) (in module azula.plugins.edm) (in module azula.plugins.eldm) (in module azula.plugins.flux) (in module azula.plugins.sana) (in module azula.plugins.sd) (in module azula.plugins.vdm)

log_prob() (azula.denoise.Gaussian method) loss() (azula.denoise.PreconditionedDenoiser method)

M

MMPSDenoiser (class in azula.guidance.mmps) module azula azula.debug azula.denoise azula.guidance azula.guidance.cfg azula.guidance.diffpir azula.guidance.dps azula.guidance.mmps azula.guidance.pgdm azula.guidance.repaint azula.guidance.tds azula.guidance.tmpd azula.hub azula.linalg azula.linalg.solve azula.nn azula.nn.attention azula.nn.embedding azula.nn.layers azula.nn.unet azula.nn.utils azula.noise azula.plugins azula.plugins.adm azula.plugins.edm azula.plugins.eldm azula.plugins.flux azula.plugins.sana azula.plugins.sd azula.plugins.utils azula.plugins.vdm azula.sample

MultiheadSelfAttention (class in azula.nn.attention)

P

Patchify() (in module azula.nn.layers) PCSampler (class in azula.sample)

PGDMSampler (class in azula.guidance.pgdm) PreconditionedDenoiser (class in azula.denoise)

R

RaiseMock (class in azula.debug) RectifiedSchedule (class in azula.noise)

ReLU2 (class in azula.nn.layers) RePaintSampler (class in azula.guidance.repaint) RMSNorm (class in azula.nn.layers)

S

Sampler (class in azula.sample) SanaDenoiser (class in azula.plugins.sana) Schedule (class in azula.noise) set_hub_dir() (in module azula.hub)

SineEncoding (class in azula.nn.embedding) skip_init (class in azula.nn.utils) StableDenoiser (class in azula.plugins.sd) step() (azula.sample.Sampler method)

T

TDSSampler (class in azula.guidance.tds) TextEncoder (class in azula.plugins.flux) (class in azula.plugins.sana) (class in azula.plugins.sd)

TMPDenoiser (class in azula.guidance.tmpd)

U

UNet (class in azula.nn.unet)

UNetBlock (class in azula.nn.unet) Unpatchify() (in module azula.nn.layers)

V

VelocityDenoiser (class in azula.plugins.vdm)

VESchedule (class in azula.noise) VPSchedule (class in azula.noise)