From 4ef37d87d5a04bc6bb7dacee0660bba3057cc02f Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 5 Jan 2023 10:26:17 +0100 Subject: Fix --- train_dreambooth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index c658ad6..1fd86b1 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -612,7 +612,7 @@ def main(): raise ValueError("--embeddings_dir must point to an existing directory") added_tokens, added_ids = load_embeddings_from_dir(tokenizer, embeddings, embeddings_dir) - print(f"Added {len(added_tokens)} tokens from embeddings dir: {zip(added_tokens, added_ids)}") + print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}") if len(args.placeholder_token) != 0: # Convert the initializer_token, placeholder_token to ids @@ -629,7 +629,7 @@ def main(): for (new_id, init_ids) in zip(new_ids, initializer_token_ids) ] - print(f"Added {len(new_ids)} new tokens: {zip(args.placeholder_token, new_ids, init_ratios)}") + print(f"Added {len(new_ids)} new tokens: {list(zip(args.placeholder_token, new_ids, init_ratios))}") else: placeholder_token_id = [] -- cgit v1.2.3-54-g00ecf