qnguyen3 commited on
Commit
dd28a0f
1 Parent(s): 052bfc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ 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.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U'])
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
  logger = build_logger("gradio_web_server", "gradio_web_server.log")
24