mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fix: ip adapter clip selection being broken
This commit is contained in:
parent
13027891d9
commit
d284e0567a
@ -65,7 +65,7 @@ class IPAdapterInvocation(BaseInvocation):
|
|||||||
ui_order=-1,
|
ui_order=-1,
|
||||||
ui_type=UIType.IPAdapterModel,
|
ui_type=UIType.IPAdapterModel,
|
||||||
)
|
)
|
||||||
clip_vision_model: Literal["auto", "ViT-H", "ViT-G"] = InputField(
|
clip_vision_model: Literal["ViT-H", "ViT-G"] = InputField(
|
||||||
description="CLIP Vision model to use. Overrides model settings. Mandatory for checkpoint models.",
|
description="CLIP Vision model to use. Overrides model settings. Mandatory for checkpoint models.",
|
||||||
default="auto",
|
default="auto",
|
||||||
ui_order=2,
|
ui_order=2,
|
||||||
@ -96,14 +96,9 @@ class IPAdapterInvocation(BaseInvocation):
|
|||||||
ip_adapter_info = context.models.get_config(self.ip_adapter_model.key)
|
ip_adapter_info = context.models.get_config(self.ip_adapter_model.key)
|
||||||
assert isinstance(ip_adapter_info, (IPAdapterInvokeAIConfig, IPAdapterCheckpointConfig))
|
assert isinstance(ip_adapter_info, (IPAdapterInvokeAIConfig, IPAdapterCheckpointConfig))
|
||||||
|
|
||||||
if self.clip_vision_model == "auto":
|
if isinstance(ip_adapter_info, IPAdapterInvokeAIConfig):
|
||||||
if isinstance(ip_adapter_info, IPAdapterInvokeAIConfig):
|
image_encoder_model_id = ip_adapter_info.image_encoder_model_id
|
||||||
image_encoder_model_id = ip_adapter_info.image_encoder_model_id
|
image_encoder_model_name = image_encoder_model_id.split("/")[-1].strip()
|
||||||
image_encoder_model_name = image_encoder_model_id.split("/")[-1].strip()
|
|
||||||
else:
|
|
||||||
raise RuntimeError(
|
|
||||||
"You need to set the appropriate CLIP Vision model for checkpoint IP Adapter models."
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
image_encoder_model_name = CLIP_VISION_MODEL_MAP[self.clip_vision_model]
|
image_encoder_model_name = CLIP_VISION_MODEL_MAP[self.clip_vision_model]
|
||||||
|
|
||||||
|
File diff suppressed because one or more lines are too long
Loading…
x
Reference in New Issue
Block a user