This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Commit 4a72ccc

authored

Add full precision attention (#189)

* Add upcast_attention option for the crossattention * Reposition docstring * Add tests Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com>

1 parent d37a752 commit 4a72cccCopy full SHA for 4a72ccc

2 files changed

+224

-137

lines changed

generative/networks/nets
- diffusion_model_unet.py
tests
- test_diffusion_model_unet.py

2 files changed

+224

-137

lines changed

Comments

(0)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 4a72ccc

2 files changed

2 files changed

File tree

2 files changed

2 files changed

0 commit comments