aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBrad Smith <bradster@infinitewarp.com>2023-03-06 13:18:35 -0500
committerBrad Smith <bradster@infinitewarp.com>2023-03-06 13:21:42 -0500
commitd118cb6ea3f1a410b5e030519dc021eafc1d6b52 (patch)
tree7775faf967c7b03ff310f4cf51c194a0973baf1c
parentb15bc73c99e6fbbeffdbdbeab39ba30276021d4b (diff)
use lowercase name for sorting; keep `UpscalerLanczos` and `UpscalerNearest` at the start of the list with `UpscalerNone`
Co-authored-by: catboxanon <122327233+catboxanon@users.noreply.github.com>
-rw-r--r--modules/modelloader.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/modelloader.py b/modules/modelloader.py
index a7ac338c..e351d808 100644
--- a/modules/modelloader.py
+++ b/modules/modelloader.py
@@ -6,7 +6,7 @@ from urllib.parse import urlparse
from basicsr.utils.download_util import load_file_from_url
from modules import shared
-from modules.upscaler import Upscaler, UpscalerNone
+from modules.upscaler import Upscaler, UpscalerLanczos, UpscalerNearest, UpscalerNone
from modules.paths import script_path, models_path
@@ -172,5 +172,5 @@ def load_upscalers():
shared.sd_upscalers = sorted(
datas,
# Special case for UpscalerNone keeps it at the beginning of the list.
- key=lambda x: x.name if not isinstance(x.scaler, UpscalerNone) else ""
+ key=lambda x: x.name.lower() if not isinstance(x.scaler, (UpscalerNone, UpscalerLanczos, UpscalerNearest)) else ""
)