├── README.md ├── benchmark_flash_attention.py └── whisper_flash_attention ├── __init__.py ├── flash_attention.py └── modeling_whisper_flash_attention.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sanchit-gandhi/whisper-flash-attention/HEAD/README.md -------------------------------------------------------------------------------- /benchmark_flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sanchit-gandhi/whisper-flash-attention/HEAD/benchmark_flash_attention.py -------------------------------------------------------------------------------- /whisper_flash_attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sanchit-gandhi/whisper-flash-attention/HEAD/whisper_flash_attention/__init__.py -------------------------------------------------------------------------------- /whisper_flash_attention/flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sanchit-gandhi/whisper-flash-attention/HEAD/whisper_flash_attention/flash_attention.py -------------------------------------------------------------------------------- /whisper_flash_attention/modeling_whisper_flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sanchit-gandhi/whisper-flash-attention/HEAD/whisper_flash_attention/modeling_whisper_flash_attention.py --------------------------------------------------------------------------------