From 56edf85c8b80d49c998bcf26392cce50d552137a Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 31 Dec 2022 23:09:41 +0100 Subject: Update --- train_ti.py | 1 + 1 file changed, 1 insertion(+) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 3776eb2..19348e5 100644 --- a/train_ti.py +++ b/train_ti.py @@ -535,6 +535,7 @@ def main(): ] new_tokens = tokenizer.add_multi_tokens(args.placeholder_token, args.num_vectors) + embeddings.resize(len(tokenizer)) for (new_token, init_ids) in zip(new_tokens, initializer_token_ids): embeddings.add_embed(new_token.ids, init_ids) -- cgit v1.2.3-54-g00ecf