From bd8ec551c960fa069482a4b4efd764f60755716b Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 27 Mar 2023 10:30:26 +0200 Subject: Fix TI --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 6c35d41..ef39c38 100644 --- a/train_ti.py +++ b/train_ti.py @@ -788,7 +788,7 @@ def main(): args.num_vectors, args.train_data_template ): - run(i, [placeholder_token], [initializer_token], [num_vectors], data_template) + run(i, [placeholder_token], [initializer_token], num_vectors, data_template) embeddings.persist() -- cgit v1.2.3-54-g00ecf