Fix for mixed case TI filenames (SDXL)

This commit is contained in:
AI-Casanova
2023-09-11 17:12:29 -05:00
committed by GitHub
parent 2f071c6586
commit d44c9d0c33
@@ -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))