a

handler.py

@@ -54,8 +54,8 @@ class EndpointHandler():
                         image=image,             
                         strength=strength,
                         num_inference_steps=num_inference_steps,
-                        denoising_start_step=denoising_start,
-                        denoising_end_step=denoising_end,
+                        denoising_start=denoising_start,
+                        denoising_end=denoising_end,
                         num_images_per_prompt=num_images_per_prompt,
                         aesthetic_score=aesthetic_score,
                         guidance_scale=guidance_scale,