sergeipetrov
/

outpainting-sdxl-vanilla-ie

Inference Endpoints

Model card Files Files and versions Community

sergeipetrov commited on Jul 19

Commit

58c4864

•

1 Parent(s): b2d28b7

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -29,7 +29,7 @@ class EndpointHandler():
         # fix the seed
         generator = torch.Generator(device="cuda").manual_seed(0)
-        image = pipe(
             prompt=prompt,
             image=image,
             mask_image=mask,

         # fix the seed
         generator = torch.Generator(device="cuda").manual_seed(0)
+        image = self.pipeline(
             prompt=prompt,
             image=image,
             mask_image=mask,