Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "5e2507f5152894b412439bf214033495e3ca20b2"
Unverified Commit 77b21628 authored by yangpei-comp's avatar yangpei-comp Committed by GitHub
Browse files

Bugfix in `pipeline_kandinsky2_2_combined.py`: Image type check mismatch (#9256) 

Update pipeline_kandinsky2_2_combined.py

Bugfix on image type check mismatch
parent 4e66513a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py
View file @ 77b21628
......@@ -547,7 +547,7 @@ class KandinskyV22Img2ImgCombinedPipeline(DiffusionPipeline):
negative_image_embeds = prior_outputs[1]
prompt = [prompt] if not isinstance(prompt, (list, tuple)) else prompt
image = [image] if isinstance(prompt, PIL.Image.Image) else image
image = [image] if isinstance(image, PIL.Image.Image) else image
if len(prompt) < image_embeds.shape[0] and image_embeds.shape[0] % len(prompt) == 0:
prompt = (image_embeds.shape[0] // len(prompt)) * prompt
......@@ -813,7 +813,7 @@ class KandinskyV22InpaintCombinedPipeline(DiffusionPipeline):
negative_image_embeds = prior_outputs[1]
prompt = [prompt] if not isinstance(prompt, (list, tuple)) else prompt
image = [image] if isinstance(prompt, PIL.Image.Image) else image
image = [image] if isinstance(image, PIL.Image.Image) else image
mask_image = [mask_image] if isinstance(mask_image, PIL.Image.Image) else mask_image
if len(prompt) < image_embeds.shape[0] and image_embeds.shape[0] % len(prompt) == 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment