From c72892835a28ae9461b70da17575237557ad744d Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 22 Feb 2023 11:38:16 -0300 Subject: [PATCH] Don't show *-np models in the list of choices --- server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server.py b/server.py index 67d74a6b..edf73158 100644 --- a/server.py +++ b/server.py @@ -469,7 +469,7 @@ def update_extensions_parameters(*kwargs): i += 1 def get_available_models(): - return sorted([item.name for item in list(Path('models/').glob('*')) if not item.name.endswith('.txt')], key=str.lower) + return sorted([item.name for item in list(Path('models/').glob('*')) if not item.name.endswith(('.txt', '-np'))], key=str.lower) def get_available_presets(): return sorted(set(map(lambda x : '.'.join(str(x.name).split('.')[:-1]), Path('presets').glob('*.txt'))), key=str.lower)