From a7b8bbc7c65a40b2be4569ae742c0c77ca39cca2 Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Thu, 21 Mar 2024 13:39:18 -0400 Subject: [PATCH] Use model_path defined in probe run --- invokeai/backend/model_manager/probe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py index e1a8cf3538..8d3818df58 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -146,11 +146,11 @@ class ModelProbe(object): raise InvalidModelConfigException(f"Unhandled combination of {format_type} and {model_type}") probe = probe_class(model_path) - + model_path = probe.model_path fields["source_type"] = fields.get("source_type") or ModelSourceType.Path fields["source"] = fields.get("source") or model_path.as_posix() fields["key"] = fields.get("key", uuid_string()) - fields["path"] = probe.model_path.as_posix() or model_path.as_posix() + fields["path"] = model_path.as_posix() fields["type"] = fields.get("type") or model_type fields["base"] = fields.get("base") or probe.get_base_type() fields["variant"] = fields.get("variant") or probe.get_variant_type()