sergeipetrov
/

outpainting-sdxl-vanilla-ie

Inference Endpoints

Model card Files Files and versions Community

sergeipetrov commited on Jul 22

Commit

796fee2

•

1 Parent(s): cb25f79

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ class EndpointHandler():
         mask = Image.open(BytesIO(base64.b64decode(inputs['mask'])))
         prompt = inputs['prompt']
         # blur mask for smooth transition
-        mask = pipeline.mask_processor.blur(mask, blur_factor=20)
         # fix the seed
         generator = torch.Generator(device="cuda").manual_seed(0)

         mask = Image.open(BytesIO(base64.b64decode(inputs['mask'])))
         prompt = inputs['prompt']
         # blur mask for smooth transition
+        mask = self.pipeline.mask_processor.blur(mask, blur_factor=20)
         # fix the seed
         generator = torch.Generator(device="cuda").manual_seed(0)