mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
tests(mm): fix test for MM, leave the UnknownModelConfig class in the list of configs
This commit is contained in:
@@ -175,7 +175,7 @@ class ModelConfigBase(ABC, BaseModel):
|
||||
if issubclass(cls, LegacyProbeMixin):
|
||||
ModelConfigBase.USING_LEGACY_PROBE.add(cls)
|
||||
# Cannot use `elif isinstance(cls, UnknownModelConfig)` because UnknownModelConfig is not defined yet
|
||||
elif cls.__name__ != "UnknownModelConfig":
|
||||
else:
|
||||
ModelConfigBase.USING_CLASSIFY_API.add(cls)
|
||||
|
||||
@staticmethod
|
||||
|
||||
Reference in New Issue
Block a user