Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add sdxl support

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-06 19:26:43 -04:00
parent d8748fd7eb
commit 7e11ff2b34
11 changed files with 390 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/textual_inversion/textual_inversion.py
+6 -2
View File
@@ -145,8 +145,12 @@ class EmbeddingDatabase:
self.word_embeddings[name] = embedding
except Exception:
self.skipped_embeddings[name] = embedding
text_inv_tokens = pipe.tokenizer.added_tokens_encoder.keys()
text_inv_tokens = [t for t in text_inv_tokens if not (len(t.split("_")) > 1 and t.split("_")[-1].isdigit())]
try:
text_inv_tokens = pipe.tokenizer.added_tokens_encoder.keys()
text_inv_tokens = [t for t in text_inv_tokens if not (len(t.split("_")) > 1 and t.split("_")[-1].isdigit())]
except Exception:
text_inv_tokens = []
pass
def load_from_file(self, path, filename):
name, ext = os.path.splitext(filename)