Update app.py

app.py

@@ -1,4 +1,4 @@
-from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
+from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL, EulerAncestralDiscreteScheduler
 from diffusers.utils import load_image
 from PIL import Image
 import torch
@@ -20,6 +20,13 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     low_cpu_mem_usage=True,
     offload_state_dict=True,
 ).to('cuda')
+pipe.scheduler = EulerAncestralDiscreteScheduler(
+    beta_start=0.00085,
+    beta_end=0.012,
+    beta_schedule="scaled_linear",
+    num_train_timesteps=1000,
+    steps_offset=1
+)
 # pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
 pipe.enable_xformers_memory_efficient_attention()
 pipe.force_zeros_for_empty_prompt = False