Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

sort models alphabetically

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/modules/textual_inversion_manager.py
View File

@ -66,8 +66,9 @@ class TextualInversionManager():
self._add_textual_inversion(embedding_info['name'], self._add_textual_inversion(embedding_info['name'],
embedding_info['embedding'], embedding_info['embedding'],
defer_injecting_tokens=defer_injecting_tokens) defer_injecting_tokens=defer_injecting_tokens)
except ValueError: except ValueError as e:
print(f' | ignoring incompatible embedding {embedding_info["name"]}') print(f' | Ignoring incompatible embedding {embedding_info["name"]}')
print(f' | The error was {str(e)}')
else: else:
print(f'>> Failed to load embedding located at {ckpt_path}. Unsupported file.') print(f'>> Failed to load embedding located at {ckpt_path}. Unsupported file.')
@ -151,7 +152,8 @@ class TextualInversionManager():
try: try:
self._inject_tokens_and_assign_embeddings(ti) self._inject_tokens_and_assign_embeddings(ti)
except ValueError as e: except ValueError as e:
print(f' | ignoring incompatible embedding trigger {ti.trigger_string}') print(f' | Ignoring incompatible embedding trigger {ti.trigger_string}')
print(f' | The error was {str(e)}')
continue continue
injected_token_ids.append(ti.trigger_token_id) injected_token_ids.append(ti.trigger_token_id)
injected_token_ids.extend(ti.pad_token_ids) injected_token_ids.extend(ti.pad_token_ids)

2
scripts/textual_inversion_fe.py
View File

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