cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

print to make sure that everything works well

Browse Source
This commit is contained in:
layerdiffusion
2024-08-04 19:03:12 -07:00
parent 9679232d29
commit 0d81ccb387
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/text_processing/classic_engine.py
+6 -2
View File
@@ -57,6 +57,7 @@ class ClassicTextProcessingEngine(torch.nn.Module):
self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape) self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape)
self.embeddings.add_embedding_dir(embedding_dir) self.embeddings.add_embedding_dir(embedding_dir)
self.embeddings.load_textual_inversion_embeddings() self.embeddings.load_textual_inversion_embeddings()
self.embedding_key = embedding_key
self.text_encoder = text_encoder self.text_encoder = text_encoder
self.tokenizer = tokenizer self.tokenizer = tokenizer
@@ -266,10 +267,13 @@ class ClassicTextProcessingEngine(torch.nn.Module):
names = [] names = []
for name, embedding in used_embeddings.items(): 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(",", "")) 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": 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 last_extra_generation_params["Emphasis"] = self.emphasis.name