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

move trigger_str registration into try block

Browse Source
This commit is contained in:
Lincoln Stein 2023-02-21 21:38:42 -05:00
parent 5e41811fb5
commit d41dcdfc46
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/modules/textual_inversion_manager.py
View File

@ -125,13 +125,13 @@ class TextualInversionManager:
embedding_info["embedding"], embedding_info["embedding"],
defer_injecting_tokens=defer_injecting_tokens, defer_injecting_tokens=defer_injecting_tokens,
) )
# remember which source file claims this trigger
self.trigger_to_sourcefile[trigger_str] = sourcefile
except ValueError as e: 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)}") print(f" | The error was {str(e)}")
# remember which source file claims this trigger
self.trigger_to_sourcefile[trigger_str] = sourcefile
def _add_textual_inversion( def _add_textual_inversion(
self, trigger_str, embedding, defer_injecting_tokens=False self, trigger_str, embedding, defer_injecting_tokens=False
) -> TextualInversion: ) -> TextualInversion: