test

app.py

@@ -19,6 +19,7 @@ from diffusers import (
     DPMSolverMultistepScheduler,
     StableDiffusionControlNetPipeline,
 )
+from diffusers.models.attention_processor import AttnProcessor2_0
 from preprocess import Preprocessor
 MAX_SEED = np.iinfo(np.int32).max
 API_KEY = os.environ.get("API_KEY", None)
@@ -95,6 +96,7 @@ if gr.NO_RELOAD:
     def init(pipe):
         pipe.enable_xformers_memory_efficient_attention()
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+        pipe.unet.set_attn_processor(AttnProcessor2_0())
         print("Model Compiled!")
     init(pipe)