From 2cac5f365d68210fa6939e4ce7bec01c6a870f28 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 2 Jul 2024 11:28:14 -0400 Subject: [PATCH] update --- modules/prompt_parser_diffusers.py | 2 +- modules/textual_inversion/textual_inversion.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index b212d08a6..167d33b53 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -389,7 +389,7 @@ def get_weighted_text_embeddings(pipe, prompt: str = "", neg_prompt: str = "", c except Exception: pooled_prompt_embeds = None negative_pooled_prompt_embeds = None - debug(f'Prompt: pooled shape={pooled_prompt_embeds[0].shape} time={(time.time() - t0):.3f}') + debug(f'Prompt: pooled shape={pooled_prompt_embeds[0].shape if pooled_prompt_embeds is not None else None} time={(time.time() - t0):.3f}') prompt_embeds = torch.cat(prompt_embeds, dim=-1) if len(prompt_embeds) > 1 else prompt_embeds[0] negative_prompt_embeds = torch.cat(negative_prompt_embeds, dim=-1) if len(negative_prompt_embeds) > 1 else \ diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 0a067ef03..21e472ec2 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -140,6 +140,8 @@ def insert_vectors(embedding, tokenizers, text_encoders, hiddensizes): this may cause collisions. """ for vector, size in zip(embedding.vec, embedding.vector_sizes): + if size not in hiddensizes: + continue idx = hiddensizes.index(size) unk_token_id = tokenizers[idx].convert_tokens_to_ids(tokenizers[idx].unk_token) if text_encoders[idx].get_input_embeddings().weight.data.shape[0] != len(tokenizers[idx]):