fix model default

handler.py

@@ -27,13 +27,13 @@ CONTROLNET_MAPPING = {
 }
 
 
-SD_ID_MAPPING = {
-    "default":  "Lykon/dreamshaper-8",
+# SD_ID_MAPPING = {
+#     "default":  "Lykon/dreamshaper-8",
 #    "dreamshaper":  "stablediffusionapi/dreamshaper-xl",
 #    "juggernaut":  "stablediffusionapi/juggernaut-xl-v8",
 #    "realistic":"SG161222/Realistic_Vision_V1.4",
 #    "rev":"s6yx/ReV_Animated"
-}
+# }
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -41,7 +41,7 @@ class EndpointHandler():
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
 
         self.stable_diffusion_id_0 = "Lykon/dreamshaper-8"
-        self.dreamshaper = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id_0,
+        self.default = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id_0,
                                                                        controlnet=self.controlnet, 
                                                                        torch_dtype=dtype,
                                                                        safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=dtype)).to("cuda")