de

app.py

@@ -45,14 +45,14 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     offload_state_dict=True,
 ).to('cuda').to(torch.float16)
 
-# pipe.scheduler = EulerAncestralDiscreteScheduler(
-#     beta_start=0.00085,
-#     beta_end=0.012,
-#     beta_schedule="scaled_linear",
-#     num_train_timesteps=1000,
-#     steps_offset=1
-# )
-pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = EulerAncestralDiscreteScheduler(
+    beta_start=0.00085,
+    beta_end=0.012,
+    beta_schedule="scaled_linear",
+    num_train_timesteps=1000,
+    steps_offset=1
+)
+# pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.force_zeros_for_empty_prompt = False
 
 def get_size(init_image):