Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -32,7 +32,7 @@ class EndpointHandler():
|
|
| 32 |
|
| 33 |
|
| 34 |
self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 35 |
-
"diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.
|
| 36 |
)
|
| 37 |
|
| 38 |
self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 35 |
+
"diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.controlnet, torch_dtype=torch.float16
|
| 36 |
)
|
| 37 |
|
| 38 |
self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
|