From d41dcdfc46c15de36a3f44c177af77f2698099ec Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Tue, 21 Feb 2023 21:38:42 -0500 Subject: [PATCH] move trigger_str registration into try block --- ldm/modules/textual_inversion_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ldm/modules/textual_inversion_manager.py b/ldm/modules/textual_inversion_manager.py index 72ae8979e2..ee86c9d226 100644 --- a/ldm/modules/textual_inversion_manager.py +++ b/ldm/modules/textual_inversion_manager.py @@ -125,13 +125,13 @@ class TextualInversionManager: embedding_info["embedding"], defer_injecting_tokens=defer_injecting_tokens, ) + # remember which source file claims this trigger + self.trigger_to_sourcefile[trigger_str] = sourcefile + except ValueError as e: print(f' | Ignoring incompatible embedding {embedding_info["name"]}') 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( self, trigger_str, embedding, defer_injecting_tokens=False ) -> TextualInversion: