update

requirements.txt

@@ -27,4 +27,4 @@ python-box
 ftfy
 librosa
 pyrubberband
-https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
+# https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

webui.py

@@ -27,6 +27,14 @@ from modules.utils.torch_opt import configure_torch_optimizations
 from modules.webui import webui_config
 from modules.webui.app import create_interface, webui_init
 
+import subprocess
+
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
+
 dcls_patch()
 ignore_useless_warnings()