From 79ef9e528cd2ae8dc49c75eeb1ea5d0d17ac9d45 Mon Sep 17 00:00:00 2001 From: manosplitsis <38662795+manosplitsis@users.noreply.github.com> Date: Fri, 4 Aug 2023 13:21:31 +0300 Subject: [PATCH] Fixed multi-token textual inversion training (#4452) * added placeholder token concatenation during training * Update examples/textual_inversion/textual_inversion.py Co-authored-by: Patrick von Platen --------- Co-authored-by: Patrick von Platen --- examples/textual_inversion/textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,