vllm/attention/backends/flash_attn.py · f7a8fa39d828136d0b8bbdd20c262602e5543ffd · OpenDAS / vllm_cscc

Stephanie Wang authored Jun 25, 2024


Signed-off-by: Stephanie Wang <swang@cs.berkeley.edu>
Signed-off-by: Stephanie <swang@anyscale.com>
Co-authored-by: Stephanie <swang@anyscale.com>

dda48115

flash_attn.py 14.4 KB