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

Fixes single-file refiner loading.
invoke-ai · Jul 25, 2024 · 9c8853f · 9c8853f
1 parent b70ac88
commit 9c8853f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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 @@ def _load_from_singlefile(
                 ModelVariantType.Normal: StableDiffusionXLPipeline,
                 ModelVariantType.Inpaint: StableDiffusionXLInpaintPipeline,
             },
+            BaseModelType.StableDiffusionXLRefiner: {
+                ModelVariantType.Normal: StableDiffusionXLPipeline,
+            },
         }
         assert isinstance(config, MainCheckpointConfig)
         try: