torch
streamlit
transformers
pillow
flash_attn==1.0.5
einops