Skip to content

Commit a369114

Browse files
fix(mm): sdxl ip adapter identification
1 parent cf0af13 commit a369114

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

invokeai/backend/model_manager/configs/ip_adapter.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,12 @@ def _get_base_or_raise(cls, mod: ModelOnDisk) -> BaseModelType:
8181
raise NotAMatchError(f"unable to determine cross attention dimension: {e}") from e
8282

8383
match cross_attention_dim:
84-
case 1280:
85-
return BaseModelType.StableDiffusionXL
8684
case 768:
8785
return BaseModelType.StableDiffusion1
8886
case 1024:
8987
return BaseModelType.StableDiffusion2
88+
case 2048:
89+
return BaseModelType.StableDiffusionXL
9090
case _:
9191
raise NotAMatchError(f"unrecognized cross attention dimension {cross_attention_dim}")
9292

@@ -154,12 +154,12 @@ def _get_base_or_raise(cls, mod: ModelOnDisk) -> BaseModelType:
154154
raise NotAMatchError(f"unable to determine cross attention dimension: {e}") from e
155155

156156
match cross_attention_dim:
157-
case 1280:
158-
return BaseModelType.StableDiffusionXL
159157
case 768:
160158
return BaseModelType.StableDiffusion1
161159
case 1024:
162160
return BaseModelType.StableDiffusion2
161+
case 2048:
162+
return BaseModelType.StableDiffusionXL
163163
case _:
164164
raise NotAMatchError(f"unrecognized cross attention dimension {cross_attention_dim}")
165165

0 commit comments

Comments
 (0)