test no coentrol net

handler.py

@@ -61,8 +61,8 @@ CONTROLNET_MAPPING = {
 class EndpointHandler():
     def __init__(self, path=""):
         # define default controlnet id and load controlnet
-        self.control_type = "depth"
-        self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
+        #self.control_type = "depth"
+        #self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
 
         #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
 
@@ -71,10 +71,10 @@ class EndpointHandler():
         #self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.stable_diffusion_id = "Lykon/dreamshaper-8"
 
-        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=dtype)).to("cuda")
+        #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=dtype)).to("cuda")
         # Define Generator with seed
         self.generator = torch.Generator(device=device.type).manual_seed(3)