cuda problem

handler.py

@@ -81,7 +81,7 @@ class EndpointHandler():
             controlnet=self.controlnet, 
             torch_dtype=dtype,
             safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
-        ).to(device)
+        )
 
                                                                       
         # Define Generator with seed