do not display the message for TI unless the list of loaded embeddings changed

1 files changed, 7 insertions, 3 deletions

diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py
index 6cf00e65..a1a406c2 100644
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@@ -112,6 +112,7 @@ class EmbeddingDatabase:
         self.skipped_embeddings = {}

         self.expected_shape = -1

         self.embedding_dirs = {}

+        self.previously_displayed_embeddings = ()

 

     def add_embedding_dir(self, path):

         self.embedding_dirs[path] = DirWithTextualInversionEmbeddings(path)

@@ -228,9 +229,12 @@ class EmbeddingDatabase:
             self.load_from_dir(embdir)

             embdir.update()

 

-        print(f"Textual inversion embeddings loaded({len(self.word_embeddings)}): {', '.join(self.word_embeddings.keys())}")

-        if len(self.skipped_embeddings) > 0:

-            print(f"Textual inversion embeddings skipped({len(self.skipped_embeddings)}): {', '.join(self.skipped_embeddings.keys())}")

+        displayed_embeddings = (tuple(self.word_embeddings.keys()), tuple(self.skipped_embeddings.keys()))

+        if self.previously_displayed_embeddings != displayed_embeddings:

+            self.previously_displayed_embeddings = displayed_embeddings

+            print(f"Textual inversion embeddings loaded({len(self.word_embeddings)}): {', '.join(self.word_embeddings.keys())}")

+            if len(self.skipped_embeddings) > 0:

+                print(f"Textual inversion embeddings skipped({len(self.skipped_embeddings)}): {', '.join(self.skipped_embeddings.keys())}")

 

     def find_embedding_at_position(self, tokens, offset):

         token = tokens[offset]