ybelkada commited on
Commit
3416529
β€’
1 Parent(s): 8337e09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ 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.3.0.post1/causal_conv1d-1.3.0.post1+cu122torch2.3cxx11abiTRUE-cp310-cp310-linux_x86_64.whl"))
10
- subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/releases/download/v1.2.2/mamba_ssm-1.2.2+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
11
 
12
  install()
13
 
 
6
 
7
  # install packages for mamba
8
  def install():
9
+ subprocess.run(shlex.split("pip install causal-conv1d mamba-ssm"))
 
10
 
11
  install()
12