File size: 160 Bytes
e292444
6497b64
e292444
 
 
6497b64
 
 
8c0556e
1
2
3
4
5
6
7
8
9
10
torch
torchvision
transformers
Pillow
gradio
# Install flash_attn directly from the GitHub repository
git+https://github.com/HazyResearch/flash-attention.git