Unverified Commit 365a9388 authored by Parag Ekbote's avatar Parag Ekbote Committed by GitHub
Browse files

Fixed Nits in Docs and Example Script (#9940)



Fixed nits in docs and example script.
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 345907f3
...@@ -497,19 +497,19 @@ class TextualInversionLoaderMixin: ...@@ -497,19 +497,19 @@ class TextualInversionLoaderMixin:
# load embeddings of text_encoder 1 (CLIP ViT-L/14) # load embeddings of text_encoder 1 (CLIP ViT-L/14)
pipeline.load_textual_inversion( pipeline.load_textual_inversion(
state_dict["clip_l"], state_dict["clip_l"],
token=["<s0>", "<s1>"], tokens=["<s0>", "<s1>"],
text_encoder=pipeline.text_encoder, text_encoder=pipeline.text_encoder,
tokenizer=pipeline.tokenizer, tokenizer=pipeline.tokenizer,
) )
# load embeddings of text_encoder 2 (CLIP ViT-G/14) # load embeddings of text_encoder 2 (CLIP ViT-G/14)
pipeline.load_textual_inversion( pipeline.load_textual_inversion(
state_dict["clip_g"], state_dict["clip_g"],
token=["<s0>", "<s1>"], tokens=["<s0>", "<s1>"],
text_encoder=pipeline.text_encoder_2, text_encoder=pipeline.text_encoder_2,
tokenizer=pipeline.tokenizer_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( pipeline.unload_textual_inversion(
tokens=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer tokens=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment