SaoYear commited on
Commit
caa9d68
·
1 Parent(s): a406eb3

so close, update torch version

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -3,7 +3,12 @@ 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==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121'])
 
 
 
 
 
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"))
 
3
  import shlex
4
  #update the system
5
  subprocess.check_call(["apt-get", "update"])
6
+ subprocess.check_call([sys.executable,"-m","pip","install",
7
+ "torch==2.2.0",
8
+ "torchvision==0.17.0",
9
+ "torchaudio==2.2.0",
10
+ "--index-url",
11
+ "https://download.pytorch.org/whl/cu121"])
12
  def install_mamba():
13
  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"))
14
  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"))