lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-05 07:29:57 +00:00
Code Issues 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

8
backend/text_processing/classic_engine.py
View File

@@ -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