karimbenharrak
commited on
Commit
•
5d55935
1
Parent(s):
58448df
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -39,7 +39,7 @@ class EndpointHandler():
|
|
39 |
)
|
40 |
|
41 |
self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
42 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
|
43 |
)
|
44 |
|
45 |
self.pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
|
|
39 |
)
|
40 |
|
41 |
self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
42 |
+
"runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
|
43 |
)
|
44 |
|
45 |
self.pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|