Jelajahi Sumber

Sort models alphabetically

oobabooga 3 tahun lalu
induk
melakukan
4b09e7e355
1 mengubah file dengan 1 tambahan dan 0 penghapusan
  1. 1 0
      server.py

+ 1 - 0
server.py

@@ -23,6 +23,7 @@ args = parser.parse_args()
 loaded_preset = None
 loaded_preset = None
 available_models = sorted(set(map(lambda x : str(x.name).replace('.pt', ''), list(Path('models/').glob('*'))+list(Path('torch-dumps/').glob('*')))))
 available_models = sorted(set(map(lambda x : str(x.name).replace('.pt', ''), list(Path('models/').glob('*'))+list(Path('torch-dumps/').glob('*')))))
 available_models = [item for item in available_models if not item.endswith('.txt')]
 available_models = [item for item in available_models if not item.endswith('.txt')]
+available_models = sorted(available_models, key=str.lower)
 available_presets = sorted(set(map(lambda x : str(x.name).split('.')[0], list(Path('presets').glob('*.txt')))))
 available_presets = sorted(set(map(lambda x : str(x.name).split('.')[0], list(Path('presets').glob('*.txt')))))
 
 
 def load_model(model_name):
 def load_model(model_name):