add safety checker with type

handler.py

@@ -85,14 +85,11 @@ class EndpointHandler():
 # safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
 # ).to(device)
 
- self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id, 
+
+ self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
  controlnet=self.controlnet, 
  torch_dtype=dtype,
- safety_checker=Node).to(device) 
-
-#StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
- 
-
+ safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=torch.float16)).to("cuda")
  # Define Generator with seed
  self.generator = torch.Generator(device=device.type).manual_seed(3)