requirements.txt

# requirements.txt
accelerate>=0.24.0
# removed_causal-conv1d>=1.1.0
datasets>=2.0.0
einops>=0.7.0
fastapi>=0.100.0
flash-attn>=2.0.0
gradio >=4.0.0
huggingface-hub>=0.19.0
mamba-ssm>=1.2.0
numpy>=1.24.0
psutil>=5.9.0 
pydantic>=2.0.0
safetensors>=0.4.0
sentencepiece>=0.1.97
tokenizers>=0.15.0
torch>=2.0.0
torch-audio>=2.0.0
transformers>=4.36.0
triton>=2.0.0
uvicorn>=0.23.0
wandb>=0.15.0