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 = AutoPipelineForText2Image.from_pretrained(self.stable_diffusion_id,
|
68 |
controlnet=self.controlnet,
|
69 |
torch_dtype=dtype,
|
70 |
safety_checker=None).to(device)
|