summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-04-09 12:17:51 +0200
committerVolpeon <git@volpeon.ink>2023-04-09 12:17:51 +0200
commit58d10c49b9c8281cbb5cefb6f59634dc4aef743b (patch)
treebd637b535b231bc81fd0915a09e638b7b53d7006
parentUpdate (diff)
downloadtextual-inversion-diff-58d10c49b9c8281cbb5cefb6f59634dc4aef743b.tar.gz
textual-inversion-diff-58d10c49b9c8281cbb5cefb6f59634dc4aef743b.tar.bz2
textual-inversion-diff-58d10c49b9c8281cbb5cefb6f59634dc4aef743b.zip
Fix
-rw-r--r--train_ti.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py
index 2d51800..ca5b113 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -811,7 +811,6 @@ def main():
811 print(f"============ TI batch {i + 1} ============") 811 print(f"============ TI batch {i + 1} ============")
812 print("") 812 print("")
813 print(stats) 813 print(stats)
814 print("")
815 814
816 filter_tokens = [token for token in args.filter_tokens if token in placeholder_tokens] 815 filter_tokens = [token for token in args.filter_tokens if token in placeholder_tokens]
817 816
@@ -874,6 +873,7 @@ def main():
874 training_iter = 1 873 training_iter = 1
875 874
876 while continue_training: 875 while continue_training:
876 print("")
877 print(f"------------ TI cycle {training_iter} ------------") 877 print(f"------------ TI cycle {training_iter} ------------")
878 print("") 878 print("")
879 879