torch
streamlit
transformers
flash-attn