Update app.py

app.py

@@ -9,6 +9,7 @@ import cv2
 import numpy as np
 from diffusers import StableDiffusionControlNetImg2ImgPipeline, ControlNetModel, DDIMScheduler
 from diffusers.models import AutoencoderKL
+from diffusers.models.attention_processor import AttnProcessor2_0
 from RealESRGAN import RealESRGAN
 import gradio as gr
 import subprocess
@@ -111,6 +112,8 @@ class ModelManager:
             progress(0, desc="Loading Stable Diffusion pipeline...")
             self.pipe = self.setup_pipeline()
             self.pipe.to(device)
+            self.pipe.unet.set_attn_processor(AttnProcessor2_0())
+            self.pipe.vae.set_attn_processor(AttnProcessor2_0())
             if USE_TORCH_COMPILE:
                 progress(0.5, desc="Compiling the model...")
                 self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)