github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Resize input image and mask for SD inpainting (#1082)

Browse Source
This commit is contained in:
jinchen62
2023-02-22 22:46:59 -08:00
committed by GitHub
parent abab59c25f
commit fab6d2e4e0
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_inpaint.py
View File

@@ -41,8 +41,10 @@ class InpaintPipeline(StableDiffusionPipeline):
super().__init__(vae, text_encoder, tokenizer, unet, scheduler)
self.vae_encode = vae_encode
def prepare_mask_and_masked_image(self, image, mask):
def prepare_mask_and_masked_image(self, image, mask, height, width):
# preprocess image
image = image.resize((width, height))
mask = mask.resize((width, height))
if isinstance(image, (Image.Image, np.ndarray)):
image = [image]
@@ -191,7 +193,7 @@ class InpaintPipeline(StableDiffusionPipeline):
# Preprocess mask and image
mask, masked_image = self.prepare_mask_and_masked_image(
image, mask_image
image, mask_image, height, width
)
# Prepare mask latent variables