karimbenharrak
/

stable-diffusion-inpainting-with-handler

StableDiffusionInpaintPipeline

stable-diffusion

stable-diffusion-diffusers

Inference Endpoints

Model card Files Files and versions Community

karimbenharrak commited on Mar 29, 2024

Commit

ad69bb7

·

verified ·

1 Parent(s): 1a72c84

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -32,7 +32,7 @@ class EndpointHandler():
         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-            "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.controlnets, torch_dtype=torch.float16
         )
         self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)

         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
+            "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.controlnet, torch_dtype=torch.float16
         )
         self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)