File size: 198 Bytes
e4e7211
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
einops
gradio
torch
transformers
numpy
sentencepiece
# triton==2.0.0.dev20221202
# -e git+https://github.com/samhavens/just-triton-flash.git#egg=flash_attn
# RuntimeError: Triton requires CUDA 11.4+