broaden regexp probe for SDXL VAEs

invoke-ai · Jun 20, 2024 · 86a7300 · 86a7300
1 parent 41b7e2a
commit 86a7300
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py
@@ -454,7 +454,7 @@ def get_base_type(self) -> BaseModelType:
         # VAEs of all base types have the same structure, so we wimp out and
         # guess using the name.
         for regexp, basetype in [
-            (r"sdxl", BaseModelType.StableDiffusionXL),
+            (r"xl", BaseModelType.StableDiffusionXL),
             (r"sd2", BaseModelType.StableDiffusion2),
             (r"vae", BaseModelType.StableDiffusion1),
         ]: