From c8dfa49d86e20bd2e18c3988b3eb48ecf6086431 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Sat, 17 Jun 2023 22:04:28 +1200
Subject: [PATCH] fix: Update missing name types to new names

---
 invokeai/backend/model_management/models/stable_diffusion.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/backend/model_management/models/stable_diffusion.py b/invokeai/backend/model_management/models/stable_diffusion.py
index 9856069ea5..0ac88c8a94 100644
--- a/invokeai/backend/model_management/models/stable_diffusion.py
+++ b/invokeai/backend/model_management/models/stable_diffusion.py
@@ -107,7 +107,7 @@ class StableDiffusion1Model(DiffusersModel):
     ) -> str:
         assert model_path == config.path
 
-        if isinstance(config, cls.CheckpointModelConfig):
+        if isinstance(config, cls.StableDiffusion1CheckpointModelConfig):
             return _convert_ckpt_and_cache(
                 version=BaseModelType.StableDiffusion1,
                 model_config=config,
@@ -220,7 +220,7 @@ class StableDiffusion2Model(DiffusersModel):
     ) -> str:
         assert model_path == config.path
 
-        if isinstance(config, cls.CheckpointModelConfig):
+        if isinstance(config, cls.StableDiffusion2CheckpointModelConfig):
             return _convert_ckpt_and_cache(
                 version=BaseModelType.StableDiffusion2,
                 model_config=config,