deepspeed>=0.14.2
transformers>=4.38.0
streamlit
flash_attn