fix(backend): add refiner to single-file load_classes

Fixes single-file refiner loading.
2024-08-30 20:32:17 +00:00 · 2024-07-25 19:01:17 +10:00 · 2024-07-25 19:01:17 +10:00 · 74cef38bcf
commit 74cef38bcf
parent bb876b8d4e
1 changed files with 3 additions and 0 deletions
--- a/invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py
+++ b/invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py
@ -98,6 +98,9 @@ class StableDiffusionDiffusersModel(GenericDiffusersLoader):
                ModelVariantType.Normal: StableDiffusionXLPipeline,
                ModelVariantType.Inpaint: StableDiffusionXLInpaintPipeline,
            },
+            BaseModelType.StableDiffusionXLRefiner: {
+                ModelVariantType.Normal: StableDiffusionXLPipeline,
+            },
        }
        assert isinstance(config, MainCheckpointConfig)
        try: