From d8450033eaa165ebeee8f006752344e1e63c25a2 Mon Sep 17 00:00:00 2001 From: Billy Date: Fri, 21 Mar 2025 17:46:18 +1100 Subject: [PATCH] Fix --- invokeai/backend/model_manager/config.py | 2 +- tests/test_model_probe.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/invokeai/backend/model_manager/config.py b/invokeai/backend/model_manager/config.py index df29b16ee5..3a9db9a2b5 100644 --- a/invokeai/backend/model_manager/config.py +++ b/invokeai/backend/model_manager/config.py @@ -402,7 +402,7 @@ class ModelConfigBase(ABC, BaseModel): cls.cast_overrides(overrides) fields.update(overrides) - type = fields.get("base") or cls.model_fields["type"].default + type = fields.get("type") or cls.model_fields["type"].default base = fields.get("base") or cls.model_fields["base"].default fields["path"] = mod.path.as_posix() diff --git a/tests/test_model_probe.py b/tests/test_model_probe.py index d075a8ed4a..8d7bb8b227 100644 --- a/tests/test_model_probe.py +++ b/tests/test_model_probe.py @@ -164,7 +164,8 @@ def test_regression_against_model_probe(datadir: Path, override_model_loading): pass if legacy_config and new_config: - assert legacy_config == new_config + assert type(legacy_config) is type(new_config) + assert legacy_config.model_dump_json() == new_config.model_dump_json() elif legacy_config: assert type(legacy_config) in ModelConfigBase._USING_LEGACY_PROBE