Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -61,7 +61,8 @@ class EndpointHandler():
|
|
| 61 |
self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
|
| 62 |
|
| 63 |
# Load StableDiffusionControlNetPipeline
|
| 64 |
-
self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
|
|
|
|
| 65 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
|
| 66 |
controlnet=self.controlnet,
|
| 67 |
torch_dtype=dtype,
|
|
|
|
| 61 |
self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
|
| 62 |
|
| 63 |
# Load StableDiffusionControlNetPipeline
|
| 64 |
+
#self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
|
| 65 |
+
self.stable_diffusion_id = "Lykon/dreamshaper-8"
|
| 66 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
|
| 67 |
controlnet=self.controlnet,
|
| 68 |
torch_dtype=dtype,
|