diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py index 51d1e483ab..78dd578b2d 100644 --- a/examples/textual_inversion/textual_inversion.py +++ b/examples/textual_inversion/textual_inversion.py @@ -708,7 +708,7 @@ def main(): data_root=args.train_data_dir, tokenizer=tokenizer, size=args.resolution, - placeholder_token=args.placeholder_token, + placeholder_token=(" ".join(tokenizer.convert_ids_to_tokens(placeholder_token_ids))), repeats=args.repeats, learnable_property=args.learnable_property, center_crop=args.center_crop,