Skip to content

Commit 02f2bbc

Browse files
author
Your Name
committed
add comment
1 parent 265ecd1 commit 02f2bbc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/loaders/textual_inversion.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,7 @@ def unload_textual_inversion(
561561
tokenizer._added_tokens_encoder[token.content] = last_special_token_id + key_id
562562
key_id += 1
563563
tokenizer._update_trie()
564+
# set correct total vocab size after removing tokens
564565
tokenizer._update_total_vocab_size()
565566

566567
# Delete from text encoder

0 commit comments

Comments
 (0)