From 9758082dc59a4b8ff164f1add1fffff3c7af670b Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Wed, 14 Feb 2024 13:16:15 -0500 Subject: [PATCH] Raise InvalidModelConfigException when unable to detect load class in ModelLoader --- invokeai/backend/model_manager/load/load_default.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/invokeai/backend/model_manager/load/load_default.py b/invokeai/backend/model_manager/load/load_default.py index 9ed0ccb2d3..1dac121a30 100644 --- a/invokeai/backend/model_manager/load/load_default.py +++ b/invokeai/backend/model_manager/load/load_default.py @@ -169,6 +169,8 @@ class ModelLoader(ModelLoaderBase): if config.get("model_type", None) == "clip_vision_model": class_name = config.get("architectures")[0] return self._hf_definition_to_type(module="transformers", class_name=class_name) + if not class_name: + raise InvalidModelConfigException("Unable to decifer Load Class based on given config.json") except KeyError as e: raise InvalidModelConfigException("An expected config.json file is missing from this model.") from e