Charles-Elena
/

ControlNet-endpoint

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

charleselena commited on Jan 19

Commit

7163d69

•

1 Parent(s): 1c7eb08

Update handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -61,7 +61,8 @@ class EndpointHandler():
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
         # Load StableDiffusionControlNetPipeline
-        self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
                                                                       controlnet=self.controlnet,
                                                                       torch_dtype=dtype,

         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
         # Load StableDiffusionControlNetPipeline
+        #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,