torch==2.5.1
transformers
gradio
sentencepiece
causal_conv1d
flash_attn
mamba_ssm
einops