Spaces:
BAAI
/
Running on L40S

File size: 241 Bytes
5afdeb1
 
261b6ba
0f8e8b9
5afdeb1
0925b75
542fa16
1
2
3
4
5
6
7
transformers==4.44.0
tiktoken==0.6.0
torch
pillow
accelerate>=0.26.0
fastapi==0.112.2
https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl