Unverified Commit 55d6453f authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[textual_inversion] use tokenizer.add_tokens to add placeholder_token (#357)

use add_tokens
parent 9ea9c6d1
...@@ -357,15 +357,18 @@ def main(): ...@@ -357,15 +357,18 @@ def main():
# Load the tokenizer and add the placeholder token as a additional special token # Load the tokenizer and add the placeholder token as a additional special token
if args.tokenizer_name: if args.tokenizer_name:
tokenizer = CLIPTokenizer.from_pretrained( tokenizer = CLIPTokenizer.from_pretrained(args.tokenizer_name)
args.tokenizer_name, additional_special_tokens=[args.placeholder_token]
)
elif args.pretrained_model_name_or_path: elif args.pretrained_model_name_or_path:
tokenizer = CLIPTokenizer.from_pretrained( tokenizer = CLIPTokenizer.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path, subfolder="tokenizer", use_auth_token=args.use_auth_token
additional_special_tokens=[args.placeholder_token], )
subfolder="tokenizer",
use_auth_token=args.use_auth_token, # Add the placeholder token in tokenizer
num_added_tokens = tokenizer.add_tokens(args.placeholder_token)
if num_added_tokens == 0:
raise ValueError(
f"The tokenizer already contains the token {args.placeholder_token}. Please pass a different"
" `placeholder_token` that is not already in the tokenizer."
) )
# Convert the initializer_token, placeholder_token to ids # Convert the initializer_token, placeholder_token to ids
......
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