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: