Kkordik commited on
Commit
36176a4
·
verified ·
1 Parent(s): ad7ddc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,8 @@ import spaces
5
  from mistral.cli.chat import load_model, generate_stream
6
 
7
  subprocess.run('pip install mamba-ssm --no-build-isolation', env={'MAMBA_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 
8
 
9
  mistral_models_path = Path.home().joinpath('mistral_models', 'mamba-codestral-7B-v0.1')
10
  mistral_models_path.mkdir(parents=True, exist_ok=True)
 
5
  from mistral.cli.chat import load_model, generate_stream
6
 
7
  subprocess.run('pip install mamba-ssm --no-build-isolation', env={'MAMBA_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
8
+ subprocess.run('pip install causal-conv1d --no-build-isolation', env={'CAUSAL_CONV1D_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
9
+
10
 
11
  mistral_models_path = Path.home().joinpath('mistral_models', 'mamba-codestral-7B-v0.1')
12
  mistral_models_path.mkdir(parents=True, exist_ok=True)