From c873d054b927850c84397ba5392e3312618d6ff4 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Thu, 9 Oct 2025 12:01:27 +1100 Subject: [PATCH] fix(mm): vae checkpoint probe checking for dir instead of file --- invokeai/backend/model_manager/configs/vae.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/configs/vae.py b/invokeai/backend/model_manager/configs/vae.py index 6c51fd770e..c171595bd6 100644 --- a/invokeai/backend/model_manager/configs/vae.py +++ b/invokeai/backend/model_manager/configs/vae.py @@ -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)