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

add batch embedding load

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-17 10:28:05 -05:00
parent 52b14442be
commit 78722065e3
4 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+1 -1
View File
@@ -137,7 +137,7 @@ def prepare_embedding_providers(pipe, clip_skip):
provider = EmbeddingsProvider(tokenizer=pipe.tokenizer, text_encoder=pipe.text_encoder, truncate=False,
returned_embeddings_type=embedding_type, device=device)
embeddings_providers.append(provider)
if hasattr(pipe, "tokenizer_2") and getattr(pipe, "text_encoder_2"):
if hasattr(pipe, "tokenizer_2") and hasattr(pipe, "text_encoder_2"):
provider = EmbeddingsProvider(tokenizer=pipe.tokenizer_2, text_encoder=pipe.text_encoder_2, truncate=False,
returned_embeddings_type=embedding_type, device=device)
embeddings_providers.append(provider)