Code

Add option to use flash attention#222

Code

Merged

Warvito merged 5 commits intomainProject-MONAI/GenerativeModels:mainfrom 210-add-option-to-use-memory-efficient-attention-to-the-diffusion-unetProject-MONAI/GenerativeModels:210-add-option-to-use-memory-efficient-attention-to-the-diffusion-unetCopy head branch name to clipboardFeb 27, 2023

+14-2

Commits

Commits on Feb 4, 2023

• Add option to use flash attention

  Show description for 1756a2c
  Warvito

  committed
  1756a2c

• Fix docstring

  Show description for 258aa66
  Warvito

  committed
  258aa66

• Add xformers==0.0.16 to requirements-dev.txt

  Show description for 9b31ca8
  Warvito

  committed
  9b31ca8

Commits on Feb 26, 2023

• Merge branch 'main' into 210-add-option-to-use-memory-efficient-attention-to-the-diffusion-unet

  Show description for ef0f743
  Warvito

  committed
  ef0f743

• Fixed torchscript error

  Show description for 617fffa
  Warvito

  committed
  617fffa