Unverified Commit d118cb6e authored by Brad Smith's avatar Brad Smith
Browse files

use lowercase name for sorting; keep `UpscalerLanczos` and `UpscalerNearest`...


use lowercase name for sorting; keep `UpscalerLanczos` and `UpscalerNearest` at the start of the list with `UpscalerNone`

Co-authored-by: default avatarcatboxanon <122327233+catboxanon@users.noreply.github.com>
parent b15bc73c
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -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 ""
    )