diff --git a/src/diffusers/loaders/textual_inversion.py b/src/diffusers/loaders/textual_inversion.py index 30098c955d6b..0162d67a340c 100644 --- a/src/diffusers/loaders/textual_inversion.py +++ b/src/diffusers/loaders/textual_inversion.py @@ -497,19 +497,19 @@ def unload_textual_inversion( # load embeddings of text_encoder 1 (CLIP ViT-L/14) pipeline.load_textual_inversion( state_dict["clip_l"], - token=["", ""], + tokens=["", ""], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer, ) # load embeddings of text_encoder 2 (CLIP ViT-G/14) pipeline.load_textual_inversion( state_dict["clip_g"], - token=["", ""], + tokens=["", ""], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2, ) - # Unload explicitly from both text encoders abd tokenizers + # Unload explicitly from both text encoders and tokenizers pipeline.unload_textual_inversion( tokens=["", ""], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer )