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>
This commit is contained in:
Brad Smith 2023-03-06 13:18:35 -05:00
parent b15bc73c99
commit d118cb6ea3
No known key found for this signature in database
GPG Key ID: CDABCFFBBD8DA710
1 changed files with 2 additions and 2 deletions

View File

@ -6,7 +6,7 @@ from urllib.parse import urlparse
from basicsr.utils.download_util import load_file_from_url from basicsr.utils.download_util import load_file_from_url
from modules import shared 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 from modules.paths import script_path, models_path
@ -172,5 +172,5 @@ def load_upscalers():
shared.sd_upscalers = sorted( shared.sd_upscalers = sorted(
datas, datas,
# Special case for UpscalerNone keeps it at the beginning of the list. # 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 ""
) )