github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(mm): vae checkpoint probe checking for dir instead of file

Browse Source
This commit is contained in:
psychedelicious
2025-10-09 12:01:27 +11:00
parent 11f03a4201
commit c873d054b9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_manager/configs/vae.py
View File

@@ -15,6 +15,7 @@ from invokeai.backend.model_manager.configs.identification_utils import (
raise_for_class_name,
raise_for_override_fields,
raise_if_not_dir,
raise_if_not_file,
state_dict_has_any_keys_starting_with,
)
from invokeai.backend.model_manager.model_on_disk import ModelOnDisk
@@ -40,7 +41,7 @@ class VAE_Checkpoint_Config_Base(Checkpoint_Config_Base):
@classmethod
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self:
raise_if_not_dir(mod)
raise_if_not_file(mod)
raise_for_override_fields(cls, override_fields)