From 8e5e4892af72062a7c3f7fd890caefdcf19bab28 Mon Sep 17 00:00:00 2001 From: Qing Date: Wed, 1 Mar 2023 09:13:23 +0800 Subject: [PATCH] fix seed generator --- lama_cleaner/model/instruct_pix2pix.py | 3 +-- lama_cleaner/model/paint_by_example.py | 3 +-- lama_cleaner/model/sd.py | 3 +-- 3 files changed, 3 insertions(+), 6 deletions(-) diff --git a/lama_cleaner/model/instruct_pix2pix.py b/lama_cleaner/model/instruct_pix2pix.py index dc57763..20e24f3 100644 --- a/lama_cleaner/model/instruct_pix2pix.py +++ b/lama_cleaner/model/instruct_pix2pix.py @@ -52,8 +52,6 @@ class InstructPix2Pix(DiffusionInpaintModel): return: BGR IMAGE edit = pipe(prompt, image=image, num_inference_steps=20, image_guidance_scale=1.5, guidance_scale=7).images[0] """ - set_seed(config.sd_seed) - output = self.model( image=PIL.Image.fromarray(image), prompt=config.prompt, @@ -62,6 +60,7 @@ class InstructPix2Pix(DiffusionInpaintModel): image_guidance_scale=config.p2p_image_guidance_scale, guidance_scale=config.p2p_guidance_scale, output_type="np.array", + generator=torch.manual_seed(config.sd_seed) ).images[0] output = (output * 255).round().astype("uint8") diff --git a/lama_cleaner/model/paint_by_example.py b/lama_cleaner/model/paint_by_example.py index d28b275..4b5ae74 100644 --- a/lama_cleaner/model/paint_by_example.py +++ b/lama_cleaner/model/paint_by_example.py @@ -51,14 +51,13 @@ class PaintByExample(DiffusionInpaintModel): mask: [H, W, 1] 255 means area to repaint return: BGR IMAGE """ - set_seed(config.paint_by_example_seed) - output = self.model( image=PIL.Image.fromarray(image), mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"), example_image=config.paint_by_example_example_image, num_inference_steps=config.paint_by_example_steps, output_type='np.array', + generator=torch.manual_seed(config.paint_by_example_seed) ).images[0] output = (output * 255).round().astype("uint8") diff --git a/lama_cleaner/model/sd.py b/lama_cleaner/model/sd.py index 7239988..71618ce 100644 --- a/lama_cleaner/model/sd.py +++ b/lama_cleaner/model/sd.py @@ -119,8 +119,6 @@ class SD(DiffusionInpaintModel): self.model.scheduler = scheduler - set_seed(config.sd_seed) - if config.sd_mask_blur != 0: k = 2 * config.sd_mask_blur + 1 mask = cv2.GaussianBlur(mask, (k, k), 0)[:, :, np.newaxis] @@ -138,6 +136,7 @@ class SD(DiffusionInpaintModel): callback=self.callback, height=img_h, width=img_w, + generator=torch.manual_seed(config.sd_seed) ).images[0] output = (output * 255).round().astype("uint8")