SaoYear commited on
Commit
8bc036f
·
1 Parent(s): af22dfc

update mamba version

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,13 +1,14 @@
1
  import subprocess
2
  import sys
3
-
4
  #update the system
5
  subprocess.check_call(["apt-get", "update"])
6
  subprocess.check_call([sys.executable,"-m","pip","install", 'torch'])
7
- subprocess.check_call([sys.executable,"-m","pip","install",'mamba-ssm'])
 
 
8
 
9
- # !pip install causal-conv1d>=1.1.0
10
- # !pip install mamba-ssm
11
 
12
  import torch
13
  import spaces
 
1
  import subprocess
2
  import sys
3
+ import shlex
4
  #update the system
5
  subprocess.check_call(["apt-get", "update"])
6
  subprocess.check_call([sys.executable,"-m","pip","install", 'torch'])
7
+ def install_mamba():
8
+ 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"))
9
+ 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"))
10
 
11
+ install_mamba()
 
12
 
13
  import torch
14
  import spaces