From 902e26507d820da0eede4f994ef0e538bb37cb66 Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Fri, 15 Mar 2024 12:49:11 -0400 Subject: [PATCH] Allow type field to be a string --- invokeai/backend/model_manager/probe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py index af26d0a985..77f2430454 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -132,7 +132,7 @@ class ModelProbe(object): format_type = ModelFormat.Diffusers if model_path.is_dir() else ModelFormat.Checkpoint model_info = None - model_type = fields['type'] if 'type' in fields else None + model_type = ModelType(fields['type']) if 'type' in fields else None if not model_type: if format_type is ModelFormat.Diffusers: model_type = cls.get_model_type_from_folder(model_path)