Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

more blend fixes

Browse Source
This commit is contained in:
Damian at mba 2022-10-21 04:23:19 +02:00
parent 404d59b1b8
commit d965540103
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/invoke/conditioning.py
View File

@ -52,10 +52,10 @@ def get_uc_and_c_and_ec(prompt_string_uncleaned, model, log_tokens=False, skip_n
blend: Blend = parsed_prompt blend: Blend = parsed_prompt
embeddings_to_blend = None embeddings_to_blend = None
for flattened_prompt in blend.prompts: for flattened_prompt in blend.prompts:
this_embedding = build_embeddings_and_tokens_for_flattened_prompt(model, flattened_prompt) this_embedding, _ = build_embeddings_and_tokens_for_flattened_prompt(model, flattened_prompt)
embeddings_to_blend = this_embedding if embeddings_to_blend is None else torch.cat( embeddings_to_blend = this_embedding if embeddings_to_blend is None else torch.cat(
(embeddings_to_blend, this_embedding)) (embeddings_to_blend, this_embedding))
conditioning, _ = WeightedFrozenCLIPEmbedder.apply_embedding_weights(embeddings_to_blend.unsqueeze(0), conditioning = WeightedFrozenCLIPEmbedder.apply_embedding_weights(embeddings_to_blend.unsqueeze(0),
blend.weights, blend.weights,
normalize=blend.normalize_weights) normalize=blend.normalize_weights)
else: else: