Daemontatox commited on
Commit
7246acf
·
verified ·
1 Parent(s): 13880c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,3 +1,10 @@
 
 
 
 
 
 
 
1
  import os
2
  import re
3
  import time
 
1
+ subprocess.run(
2
+ 'pip install flash-attn --no-build-isolation',
3
+ env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"},
4
+ shell=True
5
+ )
6
+
7
+
8
  import os
9
  import re
10
  import time