huggingface_hub
flash-attn
einops
sentence-transformers
torch==2.2.0
transformers