Update handler.py

handler.py

@@ -33,19 +33,19 @@ class EndpointHandler():
         )
 
         # load the scheduler
-        #scheduler = DPMSolverMultistepScheduler.from_pretrained(
-        #    "stabilityai/stable-diffusion-xl-base-1.0",
-        #    subfolder="scheduler",
-        #    use_lu_lambdas=True,
-        #    euler_at_final=True,
-        #)
+        scheduler = DPMSolverMultistepScheduler.from_pretrained(
+            "stabilityai/stable-diffusion-xl-base-1.0",
+            subfolder="scheduler",
+            use_lu_lambdas=True,
+            euler_at_final=True,
+        )
 
         # instantiate HF pipeline to combine all the components
         self.pipeline = StableDiffusionXLAdapterPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-base-1.0",
             adapter=adapter,
             vae=vae,
-            #scheduler=scheduler,
+            scheduler=scheduler,
             torch_dtype=torch.float16,
             variant="fp16",
             use_safetensors=True,
@@ -61,8 +61,6 @@ class EndpointHandler():
             use_safetensors=True,
         ).to("cuda")
 
-        self.pipeline.unet = torch.compile(self.pipeline.unet, mode="reduce-overhead", fullgraph=True)
-        self.refiner.unet = torch.compile(self.refiner.unet, mode="reduce-overhead", fullgraph=True)
         self.pipeline.enable_model_cpu_offload()
         self.refiner.enable_model_cpu_offload()