Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -61,7 +61,7 @@ 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/v1-5
|
65 |
"
|
66 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
|
67 |
controlnet=self.controlnet,
|
|
|
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 |
"
|
66 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
|
67 |
controlnet=self.controlnet,
|