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 26

Commit

3c30bb4

•

1 Parent(s): 135d9ff

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -29,7 +29,7 @@ class EndpointHandler():
         )
         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-            "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
         )
         self.pipe.scheduler = DDIMScheduler.from_config(self.pipe.scheduler.config)

         )
         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
+            "runwayml/stable-diffusion-inpainting", controlnet=self.controlnet, torch_dtype=torch.float16
         )
         self.pipe.scheduler = DDIMScheduler.from_config(self.pipe.scheduler.config)