fix bug

app.py

@@ -9,8 +9,6 @@ print("TORCH_CUDA", torch.cuda.is_available())
 # install packages for mamba
 def install():
     print("Install personal packages", flush=True)
-    # os.system("pip install causal_conv1d-1.0.0-cp310-cp310-linux_x86_64.whl")
-    # os.system("pip install mamba_ssm-1.0.1-cp310-cp310-linux_x86_64.whl")
     os.system("bash install.sh")
 
 install()

install.sh

@@ -1,2 +1,3 @@
+pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118
 pip install -e causal-conv1d
 pip install -e mamba

requirements.txt

@@ -8,5 +8,5 @@ setuptools
 timm
 transformers
 wheel
-torch --index-url https://download.pytorch.org/whl/cu118
-torchvision --index-url https://download.pytorch.org/whl/cu118
+# torch --index-url https://download.pytorch.org/whl/cu118
+# torchvision --index-url https://download.pytorch.org/whl/cu118