File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
invokeai/backend/model_manager/configs Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -81,12 +81,12 @@ def _get_base_or_raise(cls, mod: ModelOnDisk) -> BaseModelType:
81
81
raise NotAMatchError (f"unable to determine cross attention dimension: { e } " ) from e
82
82
83
83
match cross_attention_dim :
84
- case 1280 :
85
- return BaseModelType .StableDiffusionXL
86
84
case 768 :
87
85
return BaseModelType .StableDiffusion1
88
86
case 1024 :
89
87
return BaseModelType .StableDiffusion2
88
+ case 2048 :
89
+ return BaseModelType .StableDiffusionXL
90
90
case _:
91
91
raise NotAMatchError (f"unrecognized cross attention dimension { cross_attention_dim } " )
92
92
@@ -154,12 +154,12 @@ def _get_base_or_raise(cls, mod: ModelOnDisk) -> BaseModelType:
154
154
raise NotAMatchError (f"unable to determine cross attention dimension: { e } " ) from e
155
155
156
156
match cross_attention_dim :
157
- case 1280 :
158
- return BaseModelType .StableDiffusionXL
159
157
case 768 :
160
158
return BaseModelType .StableDiffusion1
161
159
case 1024 :
162
160
return BaseModelType .StableDiffusion2
161
+ case 2048 :
162
+ return BaseModelType .StableDiffusionXL
163
163
case _:
164
164
raise NotAMatchError (f"unrecognized cross attention dimension { cross_attention_dim } " )
165
165
You can’t perform that action at this time.
0 commit comments