Upload wheel

app.py

@@ -5,8 +5,11 @@ import torch
 
 # install packages for mamba
 def install_mamba():
+    subprocess.run(shlex.split("pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118"))
     subprocess.run(shlex.split("pip install https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.4.0/causal_conv1d-1.4.0+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
     subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/releases/download/v2.2.2/mamba_ssm-2.2.2+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
+    subprocess.run(shlex.split("pip install numpy==1.26.4"))
+
 
 install_mamba()
 

requirements.txt

@@ -8,6 +8,3 @@ pesq
 einops
 flash_attn==2.5.7
 numpy==1.26.4
-torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118
-numpy==1.26.4
-numpy==1.26.4