Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
7af2ba0
1
Parent(s):
48070ab
Upload wheel
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import torch
|
|
6 |
# install packages for mamba
|
7 |
def install_mamba():
|
8 |
subprocess.run(shlex.split("pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118"))
|
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.3cxx11abiFALSE-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.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
|
11 |
subprocess.run(shlex.split("pip install numpy==1.26.4"))
|
12 |
|
|
|
6 |
# install packages for mamba
|
7 |
def install_mamba():
|
8 |
subprocess.run(shlex.split("pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118"))
|
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.3cxx11abiFALSE-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.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
|
11 |
subprocess.run(shlex.split("pip install numpy==1.26.4"))
|
12 |
|