Skip to content

Commit 4070f26

Browse files
tests(mm): fix test for MM, leave the UnknownModelConfig class in the list of configs
1 parent 15e5c9a commit 4070f26

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

invokeai/backend/model_manager/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def __init_subclass__(cls, **kwargs):
175175
if issubclass(cls, LegacyProbeMixin):
176176
ModelConfigBase.USING_LEGACY_PROBE.add(cls)
177177
# Cannot use `elif isinstance(cls, UnknownModelConfig)` because UnknownModelConfig is not defined yet
178-
elif cls.__name__ != "UnknownModelConfig":
178+
else:
179179
ModelConfigBase.USING_CLASSIFY_API.add(cls)
180180

181181
@staticmethod

0 commit comments

Comments
 (0)