Update app.py

app.py

@@ -8,6 +8,7 @@ from PIL import Image
 from RealESRGAN import RealESRGAN
 import cv2
 import numpy as np
+from diffusers.models.attention_processor import AttnProcessor2_0
 
 # Constants
 SD15_WEIGHTS = "weights"
@@ -144,7 +145,7 @@ pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
 # Move the pipeline to the device and enable memory efficient attention
 pipe = pipe.to(device)
-pipe.enable_xformers_memory_efficient_attention()
+pipe.unet.set_attn_processor(AttnProcessor2_0())
 
 # Enable FreeU
 pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)