1 files changed, 1 insertions, 1 deletions

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()