Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -64,7 +64,7 @@ class EndpointHandler():
|
|
| 64 |
|
| 65 |
# Load StableDiffusionControlNetPipeline
|
| 66 |
self.stable_diffusion_id = "stablediffusionapi/disney-pixar-cartoon"
|
| 67 |
-
self.pipe =
|
| 68 |
controlnet=self.controlnet,
|
| 69 |
torch_dtype=dtype,
|
| 70 |
safety_checker=None).to(device)
|
|
|
|
| 64 |
|
| 65 |
# Load StableDiffusionControlNetPipeline
|
| 66 |
self.stable_diffusion_id = "stablediffusionapi/disney-pixar-cartoon"
|
| 67 |
+
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
|
| 68 |
controlnet=self.controlnet,
|
| 69 |
torch_dtype=dtype,
|
| 70 |
safety_checker=None).to(device)
|