a

handler.py

@@ -32,12 +32,10 @@ class EndpointHandler():
         num_inference_steps = data.pop("num_inference_steps", 25)
         guidance_scale = data.pop("guidance_scale", 7.5)
         negative_prompt = data.pop("negative_prompt", None)
-        height = data.pop("height", None)
-        width = data.pop("width", None)
 
         strength = data.pop("strength", 0.7)
         denoising_start = data.pop("denoising_start_step", 0)
-        denoising_end = data.pop("denoising_start_step", 0)
+        denoising_end = data.pop("denoising_end_step", 1)
         num_images_per_prompt = data.pop("num_images_per_prompt", 1)
         aesthetic_score = data.pop("aesthetic_score", 0.6)
 
@@ -48,7 +46,8 @@ class EndpointHandler():
         else:
             image = None
             
-        
+        print(f"Prompt: {inputs}, strength: {strength}, inf steps: {num_inference_steps}, denoise start: {denoising_start}, denoise_end: {denoising_end}")
+        print(f"Imgs per prompt: {num_images_per_prompt}, aesthetic_score: {aesthetic_score}, guidance_scale: {guidance_scale}, negative_prompt: {negative_prompt}")
         # run inference pipeline
         out = self.pipe(inputs, 
                         image=image,