Update handler.py

handler.py

@@ -89,7 +89,7 @@ class EndpointHandler():
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
                                                                       controlnet=self.controlnet, 
                                                                       torch_dtype=dtype,
-                                                                      safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=torch.float16)).to("cuda")
+                                                                      safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=dtype)).to("cuda")
         # Define Generator with seed
         self.generator = torch.Generator(device=device.type).manual_seed(3)
 
@@ -122,7 +122,7 @@ class EndpointHandler():
         negative_prompt = data.pop("negative_prompt", None)
         height = data.pop("height", None)
         width = data.pop("width", None)
-        controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 0.8)
+        controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
         
         # process image
         image = self.decode_base64_image(image)