From 9c8853ffd50572a21b3d18b67f12350143ba9e73 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 25 Jul 2024 19:01:17 +1000
Subject: [PATCH] fix(backend): add refiner to single-file `load_classes`

Fixes single-file refiner loading.
---
 .../model_manager/load/model_loaders/stable_diffusion.py       | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py b/invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py
index d90352f0e61..33ce4abc4d4 100644
--- 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: