From 16664da5b6689ae6cbd61e4fee932a00019acec8 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 14 Sep 2023 23:49:02 -0400 Subject: [PATCH] black --- invokeai/backend/model_management/model_probe.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/model_management/model_probe.py b/invokeai/backend/model_management/model_probe.py index 354ec0c675..c9a55f3888 100644 --- a/invokeai/backend/model_management/model_probe.py +++ b/invokeai/backend/model_management/model_probe.py @@ -511,7 +511,9 @@ class ControlNetFolderProbe(FolderProbeBase): else ( BaseModelType.StableDiffusion2 if dimension == 1024 - else BaseModelType.StableDiffusionXL if dimension == 2048 else None + else BaseModelType.StableDiffusionXL + if dimension == 2048 + else None ) ) if not base_model: