ybelkada commited on
Commit
cf7a290
·
verified ·
1 Parent(s): 0545a22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -4,13 +4,10 @@ import shlex
4
  import spaces
5
  import torch
6
 
7
- print(torch.__version__)
8
- exit(0)
9
-
10
  # install packages for mamba
11
  def install():
12
- subprocess.run(shlex.split("pip install https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.4.0/causal_conv1d-1.4.0+cu122torch2.4cxx11abiTRUE-cp310-cp310-linux_x86_64.whl"))
13
- subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/releases/download/v2.2.2/mamba_ssm-2.2.2+cu122torch2.4cxx11abiTRUE-cp39-cp39-linux_x86_64.whl"))
14
 
15
  install()
16
 
 
4
  import spaces
5
  import torch
6
 
 
 
 
7
  # install packages for mamba
8
  def install():
9
+ subprocess.run(shlex.split("pip install https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.4.0/causal_conv1d-1.4.0+cu122torch2.3cxx11abiTRUE-cp310-cp310-linux_x86_64.whl"))
10
+ subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/releases/download/v2.2.2/mamba_ssm-2.2.2+cu122torch2.3cxx11abiTRUE-cp310-cp310-linux_x86_64.whl"))
11
 
12
  install()
13