Update app.py (#3)

- Update app.py (cede6a84554808efb11346a22ea8abb8be0a9b8a)


Co-authored-by: 1inkus <[email protected]>

app.py

@@ -16,6 +16,7 @@ alloc_conf_parts = [
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = ','.join(alloc_conf_parts)
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
 os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
+import subprocess
 
 import random
 import uuid
@@ -103,6 +104,12 @@ ip_ckpt = os.path.join(local_folder2, "ip-adapter_sdxl_vit-h.bin") # Correct pat
 
 upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
 
+@spaces.GPU(required=True)
+def install_flashattn():
+    subprocess.run(['sh', './flashattn.sh'])
+    
+install_flashattn()
+
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     if style_name in styles:
         p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])