remove dtype

handler.py

@@ -76,17 +76,23 @@ class EndpointHandler():
 #                                                                      #safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
 #                                                                      safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
 
+#        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")
+#        ).to(device)
+
+        
         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")
-        )
+        ).to(device)
 
                                                                       
         # Define Generator with seed
-#        self.generator = torch.Generator(device="cpu").manual_seed(3)
-        self.generator = torch.Generator(device="cuda").manual_seed(3)
+        self.generator = torch.Generator(device="cpu").manual_seed(3)
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         """