diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-27 10:30:26 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-27 10:30:26 +0200 |
| commit | bd8ec551c960fa069482a4b4efd764f60755716b (patch) | |
| tree | 317464289a769a32c389c68d24239a7d26289b85 /train_ti.py | |
| parent | Revert to regular embeddings (diff) | |
| download | textual-inversion-diff-bd8ec551c960fa069482a4b4efd764f60755716b.tar.gz textual-inversion-diff-bd8ec551c960fa069482a4b4efd764f60755716b.tar.bz2 textual-inversion-diff-bd8ec551c960fa069482a4b4efd764f60755716b.zip | |
Fix TI
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
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(): | |||
| 788 | args.num_vectors, | 788 | args.num_vectors, |
| 789 | args.train_data_template | 789 | args.train_data_template |
| 790 | ): | 790 | ): |
| 791 | run(i, [placeholder_token], [initializer_token], [num_vectors], data_template) | 791 | run(i, [placeholder_token], [initializer_token], num_vectors, data_template) |
| 792 | embeddings.persist() | 792 | embeddings.persist() |
| 793 | 793 | ||
| 794 | 794 | ||