extensions/flash_attention/flash_attention_xformers_cuda.py · 8823cc483193699a7369df3a01f1b4e2fd02cb0d · OpenDAS / ColossalAI

Frank Lee authored Jan 25, 2024

* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

7cfed5f0

flash_attention_xformers_cuda.py 3.54 KB