david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ varlen

mirror of https://github.com/Dao-AILab/flash-attention

Antony Frolov 3566596ad8 Fix typo in RotaryEmbedding forward output type (#666)		1 year ago
..
__init__.py	ece539abd6 Add __init__.py files to subdirectories for installation	2 years ago
patch_embed.py	f1a73d0740 Run isort and black on python files	1 year ago
rotary.py	3566596ad8 Fix typo in RotaryEmbedding forward output type (#666)	1 year ago