Update app.py

app.py

@@ -19,6 +19,7 @@ from typing import Tuple
 import paramiko
 import datetime
 #from diffusers import DPMSolverSDEScheduler
+from diffusers.models.attention_processor import AttnProcessor2_0
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
@@ -109,7 +110,7 @@ def load_and_prepare_model():
     #pipe.unet.to(memory_format=torch.channels_last)
     #pipe.enable_vae_tiling()
     pipe.to(device=device, dtype=torch.bfloat16)
-
+    pipe.unet.set_attn_processor(AttnProcessor2_0())
     return pipe
     
 pipe = load_and_prepare_model()