Update handler.py

handler.py

@@ -45,7 +45,7 @@ class EndpointHandler():
             torch_dtype=torch.float16,
         )
         # use DPMSolverMultistepScheduler
-        self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
+        self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
 
         self.pipe.enable_model_cpu_offload()
 
@@ -130,7 +130,7 @@ class EndpointHandler():
             image=image,
             mask_image=mask_image,
             guidance_scale=guidance_scale, #8.0
-            num_inference_steps=num_inference_steps, #100
+            num_inference_steps=int(num_inference_steps/10), #100
             strength=strength, #0.2
             output_type="latent",  # let's keep in latent to save some VRAM
         ).images[0]
@@ -141,7 +141,7 @@ class EndpointHandler():
             prompt=prompt,
             image=image,
             guidance_scale=guidance_scale, #8.0
-            num_inference_steps=num_inference_steps, #100
+            num_inference_steps=int(num_inference_steps/10), #100
             strength=strength, #0.2
         ).images[0]