Leyo commited on
Commit
9e85d8f
1 Parent(s): 13fd3a2
Files changed (1) hide show
  1. app_dialogue.py +1 -1
app_dialogue.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import subprocess
3
 
4
  # Install flash attention
5
- # subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
 
7
 
8
  import copy
 
2
  import subprocess
3
 
4
  # Install flash attention
5
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
 
7
 
8
  import copy