From 0d81ccb387e2500d0944936ca4e0ea8e6c55a604 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Sun, 4 Aug 2024 19:03:12 -0700 Subject: [PATCH] print to make sure that everything works well --- backend/text_processing/classic_engine.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/backend/text_processing/classic_engine.py b/backend/text_processing/classic_engine.py index 7c16de70..10e9079f 100644 --- a/backend/text_processing/classic_engine.py +++ b/backend/text_processing/classic_engine.py @@ -57,6 +57,7 @@ class ClassicTextProcessingEngine(torch.nn.Module): self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape) self.embeddings.add_embedding_dir(embedding_dir) self.embeddings.load_textual_inversion_embeddings() + self.embedding_key = embedding_key self.text_encoder = text_encoder self.tokenizer = tokenizer @@ -266,10 +267,13 @@ class ClassicTextProcessingEngine(torch.nn.Module): names = [] for name, embedding in used_embeddings.items(): - print(f'Used Embedding: {name}') + print(f'[Textual Inversion] Used Embedding [{name}] in CLIP of [{self.embedding_key}]') names.append(name.replace(":", "").replace(",", "")) - last_extra_generation_params["TI"] = ", ".join(names) + if "TI" in last_extra_generation_params: + last_extra_generation_params["TI"] += ", " + ", ".join(names) + else: + last_extra_generation_params["TI"] = ", ".join(names) if any(x for x in texts if "(" in x or "[" in x) and self.emphasis.name != "Original": last_extra_generation_params["Emphasis"] = self.emphasis.name