We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15e5c9a commit 6bf6966Copy full SHA for 6bf6966
tests/test_model_probe.py
@@ -17,6 +17,7 @@
17
MainDiffusersConfig,
18
ModelConfigBase,
19
ModelConfigFactory,
20
+ UnknownModelConfig,
21
get_model_discriminator_value,
22
)
23
from invokeai.backend.model_manager.legacy_probe import (
@@ -257,6 +258,10 @@ def test_any_model_config_includes_all_config_classes():
257
258
extracted.add(config_class)
259
260
expected = set(ModelConfigBase.all_config_classes()) - {MinimalConfigExample}
261
+
262
+ # UnknownModelConfig is deliberately excluded from AnyModelConfig to avoid it being used for model probing.
263
+ expected.remove(UnknownModelConfig)
264
265
assert extracted == expected
266
267
0 commit comments