Kkordik commited on
Commit
aba5131
·
verified ·
1 Parent(s): 65e0386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,8 @@ from pathlib import Path
4
  import spaces
5
  from mistral.cli.chat import load_model, generate_stream
6
 
 
 
7
  mistral_models_path = Path.home().joinpath('mistral_models', 'mamba-codestral-7B-v0.1')
8
  mistral_models_path.mkdir(parents=True, exist_ok=True)
9
 
 
4
  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)
11