sort models alphabetically

This commit is contained in:
Lincoln Stein
2023-01-20 15:17:54 -05:00
parent ab675af264
commit 195294e74f
2 changed files with 6 additions and 4 deletions

View File

@ -219,7 +219,7 @@ class textualInversionForm(npyscreen.FormMultiPageAction):
def get_model_names(self)->(List[str],int):
conf = OmegaConf.load(os.path.join(Globals.root,'configs/models.yaml'))
model_names = list(conf.keys())
model_names = sorted(list(conf.keys()))
defaults = [idx for idx in range(len(model_names)) if 'default' in conf[model_names[idx]]]
return (model_names,defaults[0])