Fix typo: checkpoint_alisases

3 files changed, 9 insertions, 8 deletions

diff --git a/modules/api/api.py b/modules/api/api.py
index 224bbfc6..5793bb44 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -22,7 +22,7 @@ from modules.textual_inversion.textual_inversion import create_embedding, train_
 from modules.textual_inversion.preprocess import preprocess
 from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork
 from PIL import PngImagePlugin,Image
-from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights, checkpoint_alisases
+from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights, checkpoint_aliases
 from modules.sd_vae import vae_dict
 from modules.sd_models_config import find_checkpoint_config_near_filename
 from modules.realesrgan_model import get_realesrgan_models
@@ -519,7 +519,7 @@ class Api:
 
     def set_config(self, req: Dict[str, Any]):
         checkpoint_name = req.get("sd_model_checkpoint", None)
-        if checkpoint_name is not None and checkpoint_name not in checkpoint_alisases:
+        if checkpoint_name is not None and checkpoint_name not in checkpoint_aliases:
             raise RuntimeError(f"model {checkpoint_name!r} not found")
 
         for k, v in req.items():
diff --git a/modules/processing.py b/modules/processing.py
index 21d1492c..cd568a20 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -606,7 +606,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
 

     try:

         # if no checkpoint override or the override checkpoint can't be found, remove override entry and load opts checkpoint

-        if sd_models.checkpoint_alisases.get(p.override_settings.get('sd_model_checkpoint')) is None:

+        if sd_models.checkpoint_aliases.get(p.override_settings.get('sd_model_checkpoint')) is None:

             p.override_settings.pop('sd_model_checkpoint', None)

             sd_models.reload_model_weights()

 

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 653c4cc0..060e0007 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -23,7 +23,8 @@ model_dir = "Stable-diffusion"
 model_path = os.path.abspath(os.path.join(paths.models_path, model_dir))

 

 checkpoints_list = {}

-checkpoint_alisases = {}

+checkpoint_aliases = {}

+checkpoint_alisases = checkpoint_aliases  # for compatibility with old name

 checkpoints_loaded = collections.OrderedDict()

 

 

@@ -66,7 +67,7 @@ class CheckpointInfo:
     def register(self):

         checkpoints_list[self.title] = self

         for id in self.ids:

-            checkpoint_alisases[id] = self

+            checkpoint_aliases[id] = self

 

     def calculate_shorthash(self):

         self.sha256 = hashes.sha256(self.filename, f"checkpoint/{self.name}")

@@ -112,7 +113,7 @@ def checkpoint_tiles():
 

 def list_models():

     checkpoints_list.clear()

-    checkpoint_alisases.clear()

+    checkpoint_aliases.clear()

 

     cmd_ckpt = shared.cmd_opts.ckpt

     if shared.cmd_opts.no_download_sd_model or cmd_ckpt != shared.sd_model_file or os.path.exists(cmd_ckpt):

@@ -136,7 +137,7 @@ def list_models():
 

 

 def get_closet_checkpoint_match(search_string):

-    checkpoint_info = checkpoint_alisases.get(search_string, None)

+    checkpoint_info = checkpoint_aliases.get(search_string, None)

     if checkpoint_info is not None:

         return checkpoint_info

 

@@ -166,7 +167,7 @@ def select_checkpoint():
     """Raises `FileNotFoundError` if no checkpoints are found."""

     model_checkpoint = shared.opts.sd_model_checkpoint

 

-    checkpoint_info = checkpoint_alisases.get(model_checkpoint, None)

+    checkpoint_info = checkpoint_aliases.get(model_checkpoint, None)

     if checkpoint_info is not None:

         return checkpoint_info