Kkordik commited on
Commit
4eafd11
·
verified ·
1 Parent(s): 89011bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
 
8
  # Install required packages
9
  subprocess.run('pip install causal-conv1d --no-build-isolation', env={'CAUSAL_CONV1D_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
10
- subprocess.run('pip install mistral_inference mamba-ssm --no-build-isolation', env={'MAMBA_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
11
 
12
  # Import after installation
13
  from mistral_inference.mamba import Mamba
 
7
 
8
  # Install required packages
9
  subprocess.run('pip install causal-conv1d --no-build-isolation', env={'CAUSAL_CONV1D_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
10
+ subprocess.run('pip install mamba-ssm --no-build-isolation', env={'MAMBA_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
11
 
12
  # Import after installation
13
  from mistral_inference.mamba import Mamba