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

Use correct format type on model

Browse Source
This commit is contained in:
brandonrising 2024-03-25 20:53:35 -04:00 committed by Brandon
parent a7b8bbc7c6
commit 90f32d17ba
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -147,6 +147,7 @@ class ModelProbe(object):
probe = probe_class(model_path) probe = probe_class(model_path)
model_path = probe.model_path model_path = probe.model_path
format_type = probe.get_format()
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())
@ -160,7 +161,8 @@ class ModelProbe(object):
fields["description"] = ( fields["description"] = (
fields.get("description") or f"{fields['base'].value} {model_type.value} model {fields['name']}" fields.get("description") or f"{fields['base'].value} {model_type.value} model {fields['name']}"
) )
fields["format"] = fields.get("format") or probe.get_format() fields["format"] = fields.get("format") or format_type
fields["hash"] = fields.get("hash") or ModelHash(algorithm=hash_algo).hash(model_path) fields["hash"] = fields.get("hash") or ModelHash(algorithm=hash_algo).hash(model_path)
fields["default_settings"] = fields.get("default_settings") fields["default_settings"] = fields.get("default_settings")