Update app.py

app.py

@@ -147,19 +147,17 @@ def load_model():
     
 pipe, upscaler_2 = load_model()
 
-#fa_processor = FlashAttentionProcessor()
+fa_processor = FlashAttentionProcessor()
 
-#for name, module in pipe.transformer.named_modules():
+for name, module in pipe.transformer.named_modules():
     # Only replace processors that are instances of AttnProcessor2_0
     # This automatically includes the self-attention and cross-attention within transformer blocks.
-#    if isinstance(module, AttnProcessor2_0):
-#        module.processor = fa_processor
-
+    if isinstance(module, AttnProcessor2_0):
+        module.processor = fa_processor
 
 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)