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 1

Commit

a678ae0

•

1 Parent(s): 3f1ef06

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -55,7 +55,7 @@ class EndpointHandler():
         # run inference pipeline
         out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
-        #image = out.images[0].resize((1024, 1024))
         self.pipe2.enable_xformers_memory_efficient_attention()

         # run inference pipeline
         out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
+        image = out.images[0].resize((1024, 1024))
         self.pipe2.enable_xformers_memory_efficient_attention()