Skip to content

UNI-D²

CANDI DiT

UNI-D²

Home
Architecture
Contributing
Examples
Examples
- CANDI
- MDLM
- PGM
- BD3LM
- FlexMDM
- GIDD
- SEDD
- UDLM
Extension Guides
Extension Guides
API Reference
API Reference
- Algorithms
  Algorithms
  - Base Trainer
  - CANDI
  - MDLM
  - UDLM
  - AR
  - BD3LM
  - FlexMDM
  - GIDD
  - Partition
  - SEDD
- Models
  Models
  - DiT
  - CANDI DiT CANDI DiT
    Table of contents
    
    dit_candi
  - Common Layers
  - Block DiT
  - EMA
  - Encoder-Decoder
  - FlexMDM Transformer
  - HF GPT2
- Forward Process
  Forward Process
  - Base
  - Absorbing
  - Block Absorbing
  - CANDI Hybrid
  - Uniform
  - FlexMDM
  - Utils
- Noise Schedules
  Noise Schedules
  - Base
  - Cosine
  - Flex
  - Geometric
  - Hybrid
  - Linear
  - Log-Linear
- Sampling
  Sampling
  - Base
  - Absorbing
  - AR
  - BD3LM
  - CANDI
  - FlexMDM
  - GIDD
  - Partition
  - Uniform
- Evaluations
  Evaluations
- Callbacks
  Callbacks
  - Sample Saver
  - Training Latency

CANDI DiT

`discrete_diffusion.models.dit_candi`

DiT model architecture adapted for continuous-discrete hybrid diffusion.