From d44c9d0c33f0cc71c740b00f94f3435cb561c45b Mon Sep 17 00:00:00 2001 From: AI-Casanova <54461896+AI-Casanova@users.noreply.github.com> Date: Mon, 11 Sep 2023 17:12:29 -0500 Subject: [PATCH] Fix for mixed case TI filenames (SDXL) --- modules/textual_inversion/textual_inversion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index c8cee595d..a9e200822 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -144,9 +144,9 @@ class EmbeddingDatabase: embeddings_dict[k] = f.get_tensor(k) for i in range(len(embeddings_dict["clip_l"])): if i == 0: - token = name + token = name.lower() else: - token = f"{name}_{i}" + token = f"{name.lower()}_{i}" pipe.tokenizer.add_tokens(token) token_id = pipe.tokenizer.convert_tokens_to_ids(token) pipe.text_encoder.resize_token_embeddings(len(pipe.tokenizer))