pip
torch
gradio
transformers
sentencepiece
einops
mamba_ssm
causal_conv1d
flash_attn