Move output_count

handler.py

@@ -76,14 +76,12 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", None)
         image = data.pop("image", None)
-        output_count = data.pop("output_count", None)
+        
         controlnet_type = data.pop("controlnet_type", None)
         
         # Check if neither prompt nor image is provided
         if prompt is None and image is None:
             return {"error": "Please provide a prompt and base64 encoded image."}
-        if output_count is None:
-            return {"error": "Please provide an output_count."}
         
         # Check if a new controlnet is provided
         if controlnet_type is not None and controlnet_type != self.control_type:
@@ -101,6 +99,7 @@ class EndpointHandler():
         height = data.pop("height", None)
         width = data.pop("width", None)
         controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
+        output_count = data.pop("output_count", None)
         
         # process image
         image = self.decode_base64_image(image)