Update app.py

app.py

@@ -13,6 +13,13 @@ from diffusers.models import AutoencoderKL
 from PIL import Image
 
 # Pre-Initialize
+
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
+
 DEVICE = "auto"
 if DEVICE == "auto":
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"