qnguyen3 commited on
Commit
67b38d0
1 Parent(s): 705ce5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -18,7 +18,9 @@ from serve.constants import LOGDIR
18
  from serve.utils import (build_logger, server_error_msg, violates_moderation, moderation_msg)
19
  import subprocess
20
 
21
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 
22
 
23
  #logger = build_#logger("gradio_web_server", "gradio_web_server.log")
24
 
 
18
  from serve.utils import (build_logger, server_error_msg, violates_moderation, moderation_msg)
19
  import subprocess
20
 
21
+ # subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
22
+
23
+ subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U'])
24
 
25
  #logger = build_#logger("gradio_web_server", "gradio_web_server.log")
26