lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-25 16:59:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Text Processing Engine is Finished

Browse Source 
100% reproduce all previous results, including TI embeddings, LoRAs in CLIP, emphasize settings, BREAK, timestep swap scheduling, AB mixture, advanced uncond, etc
Backend is 85% finished
This commit is contained in:
layerdiffusion
2024-08-04 14:13:37 -07:00
parent 2791203d5b
commit a72154405e
8 changed files with 123 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/textual_inversion/textual_inversion.py
View File

@@ -127,7 +127,7 @@ class EmbeddingDatabase:
return self.register_embedding_by_name(embedding, model, embedding.name)
def register_embedding_by_name(self, embedding, model, name):
ids = model.cond_stage_model.tokenize([name])[0]
ids = [0, 0, 0] # model.cond_stage_model.tokenize([name])[0]
first_id = ids[0]
if first_id not in self.ids_lookup:
self.ids_lookup[first_id] = []
@@ -183,11 +183,7 @@ class EmbeddingDatabase:
if data is not None:
embedding = create_embedding_from_data(data, name, filename=filename, filepath=path)
if self.expected_shape == -1 or self.expected_shape == embedding.shape:
self.register_embedding(embedding, shared.sd_model)
else:
self.skipped_embeddings[name] = embedding
self.register_embedding(embedding, None)
else:
print(f"Unable to load Textual inversion embedding due to data issue: '{name}'.")