github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Only allow a single image prompt for FLUX IP-Adapters (haven't really looked into this much, but punting on it for now).

Browse Source
This commit is contained in:
Ryan Dick
2024-10-22 16:32:01 +00:00
parent 740f6eb19f
commit e48cab0276
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/invocations/flux_denoise.py
View File

@@ -570,6 +570,11 @@ class FluxDenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
else:
raise ValueError(f"Unsupported IP-Adapter image type: {type(ip_adapter_field.image)}")
if len(ipa_image_fields) != 1:
raise ValueError(
f"FLUX IP-Adapter only supports a single image prompt (received {len(ipa_image_fields)})."
)
ipa_images = [context.images.get_pil(image.image_name) for image in ipa_image_fields]
pos_images: list[npt.NDArray[np.uint8]] = []

2
invokeai/app/invocations/flux_ip_adapter.py
View File

@@ -35,7 +35,7 @@ class FluxIPAdapterInvocation(BaseInvocation):
# FLUXIPAdapterInvocation is based closely on IPAdapterInvocation, but with some unsupported features removed.
image: Union[ImageField, List[ImageField]] = InputField(description="The IP-Adapter image prompt(s).")
image: ImageField = InputField(description="The IP-Adapter image prompt(s).")
ip_adapter_model: ModelIdentifierField = InputField(
description="The IP-Adapter model.", title="IP-Adapter Model", ui_type=UIType.IPAdapterModel
)