This commit is contained in:
Vladimir Mandic
2023-09-09 19:39:21 -04:00
parent 7bda411738
commit 36001151bb
5 changed files with 14 additions and 17 deletions
+2 -1
View File
@@ -62,7 +62,8 @@ class DiffusersTextualInversionManager(BaseTextualInversionManager):
replacement += f" {token}_{i}"
i += 1
prompt = prompt.replace(token, replacement)
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]: