fix: Update missing name types to new names

invoke-ai · Jun 17, 2023 · c8dfa49 · c8dfa49
1 parent 76dd749
commit c8dfa49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/invokeai/backend/model_management/models/stable_diffusion.py b/invokeai/backend/model_management/models/stable_diffusion.py
@@ -107,7 +107,7 @@ def convert_if_required(
     ) -> 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 @@ def convert_if_required(
     ) -> 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,