Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
1f8411b
1
Parent(s):
d450f41
Test on lengths
Browse files
app.py
CHANGED
|
@@ -7,6 +7,8 @@ import shutil
|
|
| 7 |
import glob
|
| 8 |
import gradio as gr
|
| 9 |
|
|
|
|
|
|
|
| 10 |
# install packages for mamba
|
| 11 |
def install_mamba():
|
| 12 |
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"))
|
|
|
|
| 7 |
import glob
|
| 8 |
import gradio as gr
|
| 9 |
|
| 10 |
+
os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
|
| 11 |
+
|
| 12 |
# install packages for mamba
|
| 13 |
def install_mamba():
|
| 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"))
|