v1

app.py

@@ -18,6 +18,8 @@ from torchvision.transforms.functional import pil_to_tensor
 # flash attention
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+subprocess.run('pip install causal-conv1d>=1.2.0 --no-build-isolation', shell=True)
+subprocess.run('pip install mamba-ssm --no-build-isolation', shell=True)
 
 # accel
 accel = Accelerator()