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

better diffusers embedding handling

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-26 12:04:26 -04:00
parent e661d7bce7
commit 14aec521ed
5 changed files with 46 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+2 -2
View File
@@ -62,8 +62,8 @@ class DiffusersTextualInversionManager(BaseTextualInversionManager):
replacement += f" {token}_{i}"
i += 1
prompt = prompt.replace(token, replacement)
if hasattr(self.pipe, 'embedding_db'):
self.pipe.embedding_db.embeddings_used = list(set(self.pipe.embedding_db.embeddings_used))
if hasattr(self.pipe, 'embedding_db'):
self.pipe.embedding_db.embeddings_used = list(set(self.pipe.embedding_db.embeddings_used))
return prompt
def expand_textual_inversion_token_ids_if_necessary(self, token_ids: typing.List[int]) -> typing.List[int]: