Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix paths in probe

Browse Source
This commit is contained in:
Brandon Rising 2024-03-21 13:25:24 -04:00 committed by Brandon
parent 18098cc4b9
commit befcf779dc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_manager/probe.py
View File

@ -150,7 +150,7 @@ class ModelProbe(object):
fields["source_type"] = fields.get("source_type") or ModelSourceType.Path fields["source_type"] = fields.get("source_type") or ModelSourceType.Path
fields["source"] = fields.get("source") or model_path.as_posix() fields["source"] = fields.get("source") or model_path.as_posix()
fields["key"] = fields.get("key", uuid_string()) fields["key"] = fields.get("key", uuid_string())
fields["path"] = model_path.as_posix() fields["path"] = probe.model_path.as_posix() or model_path.as_posix()
fields["type"] = fields.get("type") or model_type fields["type"] = fields.get("type") or model_type
fields["base"] = fields.get("base") or probe.get_base_type() fields["base"] = fields.get("base") or probe.get_base_type()
fields["variant"] = fields.get("variant") or probe.get_variant_type() fields["variant"] = fields.get("variant") or probe.get_variant_type()
@ -652,7 +652,7 @@ class TextualInversionFolderProbe(TextualInversionCheckpointProbe):
raise InvalidModelConfigException( raise InvalidModelConfigException(
f"Unable to determine base type for {model_path}: expected exactly one valid model file, found {[f.name for f in files]}." f"Unable to determine base type for {model_path}: expected exactly one valid model file, found {[f.name for f in files]}."
) )
super().__init__(model_path) super().__init__(files.pop())
class ONNXFolderProbe(PipelineFolderProbe): class ONNXFolderProbe(PipelineFolderProbe):