Repository navigation

Add RAE Diffusion Transformer - #13231

Open

plugyawn wants to merge 32 commits into

huggingface:mainfrom

plugyawn:rae-dit-training

Commits on Mar 30, 2026
- Add Stage-2 RAE DiT model, pipeline, and tooling
  plugyawn
  committed
  6957708
- Add RAE DiT resume-order verifier
  plugyawn
  committed
  d5bf0d9
- Add RAE DiT training smoke test
  plugyawn
  committed
  f422041
- Sync RAE DiT stack with diffusers quality checks
  plugyawn
  committed
  626a89d
- Add RAE DiT API docs
  plugyawn
  committed
  9c8f052
- Rename RAEDiTTransformer2DModel to RAEDiT2DModel
  plugyawn
  committed
  8314f0e
- Fix RAE DiT review regressions
  plugyawn
  committed
  aa47b26
- Remove RAE DiT validation helper scripts from PR
  plugyawn
  committed
  a2506ff
- Add RAE DiT training validation sampling
  plugyawn
  committed
  bc1b237
- Align RAE DiT with diffusers patterns
  plugyawn
  committed
  054ae73
- Localize RAE loading and drop unused guidance transformer
  plugyawn
  committed
  c871658
- Localize RAE nested init loading
  plugyawn
  committed
  fa2d665
- Fix RAEDiT projector init and dtype handling
  plugyawn
  committed
  34d6351
- Harden RAE DiT training schedule helpers
  plugyawn
  committed
  afc2db7
- Harden RAE DiT conversion and pipeline helpers
  plugyawn
  committed
  b9d46ca
- Merge branch 'main' into rae-dit-training
  kashif
  authored
  9354b55
Commits on Apr 19, 2026
- Merge branch 'main' into rae-dit-training
  kashif
  authored
  b16c31f
Commits on Apr 20, 2026
- Fix RAEDiT scheduler and conditioning edge cases
  plugyawn
  committed
  c9770f6
Commits on May 1, 2026
- Add RAEDiT attention fusion support
  plugyawn
  committed
  a98c667
Commits on May 2, 2026
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  b6aad13
Commits on May 18, 2026
- Fix RAEDiT pipeline export pos embed preservation
  plugyawn
  committed
  49465e6
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  337737a
Commits on May 29, 2026
- Merge branch 'main' into rae-dit-training
  kashif
  authored
  ae54f9f
Commits on Jun 3, 2026
- Merge branch 'main' into rae-dit-training
  kashif
  authored
  db19440
- fix rae encoder init under no_init_weights
  kashif
  committed
  1ed9c00
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  4fc8543
Commits on Jun 4, 2026
- Address RAEDiT pipeline review follow-ups
  plugyawn
  committed
  7bf8865
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  1b48fba
- Fix RAEDiT quality checks
  plugyawn
  committed
  505c5b2
Commits on Jun 9, 2026
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  babc1ed
Commits on Jun 12, 2026
- Merge branch 'main' into rae-dit-training
  plugyawn
  authored
  6b9c641
Commits on Aug 1, 2026
- Merge branch 'main' into rae-dit-training
  kashif
  authored
  e3d0654

Commits

Commits on Mar 30, 2026

Commits on Apr 19, 2026

Commits on Apr 20, 2026

Commits on May 1, 2026

Commits on May 2, 2026

Commits on May 18, 2026

Commits on May 29, 2026

Commits on Jun 3, 2026

Commits on Jun 4, 2026

Commits on Jun 9, 2026

Commits on Jun 12, 2026

Commits on Aug 1, 2026