Code

Add full precision attention#189

Code

Merged

Warvito merged 3 commits intomainProject-MONAI/GenerativeModels:mainfrom 187-allow-to-force-to-use-full-precision-when-computing-attentionProject-MONAI/GenerativeModels:187-allow-to-force-to-use-full-precision-when-computing-attentionCopy head branch name to clipboardJan 23, 2023

+224-137

Commits

Commits on Jan 14, 2023

• Add upcast_attention option for the crossattention

  Show description for fc4d6c8
  Warvito

  committed
  fc4d6c8

  Copy full SHA for fc4d6c8

• Reposition docstring

  Show description for 312d4ab
  Warvito

  committed
  312d4ab

  Copy full SHA for 312d4ab

• Add tests

  Show description for 62deae0
  Warvito

  committed
  62deae0

  Copy full SHA for 62deae0