Repository navigation

Fix F16 Usage in DiffusionUnet - #9252

Open

blakedewey wants to merge 5 commits into

Project-MONAI:devfrom

blakedewey:f16_stability_diffusion

Open

Fix F16 Usage in DiffusionUnet#9252
blakedewey wants to merge 5 commits into
Project-MONAI:devfrom
blakedewey:f16_stability_diffusion

Commits on Apr 29, 2026
- Ensure f32 upcast for attention and time embedding
  blakedewey
  committed
  7bfbc2b
- Add flag to upcast embeddings in DiffusionModelUNet
  blakedewey
  committed
  b11ad90
- Add v to upcast block, recast to input_dtype after attention
  blakedewey
  committed
  01fed76
Commits on Apr 30, 2026
- add flag to cast after norm
  blakedewey
  committed
  4af825c
Commits on Nov 5, 2026
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  2248c75