Locutusque commited on
Commit
17f1f96
·
verified ·
1 Parent(s): c6b7ad3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ import spaces
6
  import os
7
 
8
 
9
- # Install flash-attn
10
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
11
  # Initialize the model pipeline
12
  generator = pipeline('text-generation', model='hydra-project/ChatHercules-2.5-Mistral-7B', torch_dtype=torch.bfloat16)
13
  @spaces.GPU
 
6
  import os
7
 
8
 
9
+
 
10
  # Initialize the model pipeline
11
  generator = pipeline('text-generation', model='hydra-project/ChatHercules-2.5-Mistral-7B', torch_dtype=torch.bfloat16)
12
  @spaces.GPU