From aec567179d2e92da002c026bbd11c0b9a50f0477 Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Sun, 17 Mar 2024 22:43:43 -0400 Subject: [PATCH] Pull format type setting out of model_type if statement --- 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 e814644a45..d43b160750 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -138,7 +138,7 @@ class ModelProbe(object): model_type = cls.get_model_type_from_folder(model_path) else: model_type = cls.get_model_type_from_checkpoint(model_path) - format_type = ModelFormat.ONNX if model_type == ModelType.ONNX else format_type + format_type = ModelFormat.ONNX if model_type == ModelType.ONNX else format_type probe_class = cls.PROBES[format_type].get(model_type) if not probe_class: