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)