Update app.py

app.py

@@ -21,6 +21,14 @@ import io
 from google.oauth2 import service_account
 from google.cloud import storage
 
+
+@spaces.GPU(required=True)
+def install_flashattn():
+    subprocess.run(['sh', './flashattn.sh'])
+
+install_flashattn()
+
+
 import torch
 # Torch performance settings
 torch.backends.cuda.matmul.allow_tf32 = False
@@ -88,6 +96,7 @@ upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(devic
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
 
+
 @spaces.GPU(duration=45)
 def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width, height, guidance, steps, progress=gr.Progress(track_tqdm=True)):
     seed = random.randint(0, MAX_SEED)