summaryrefslogtreecommitdiffstats
path: root/train_ti.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-03-31 13:57:07 +0200
committerVolpeon <git@volpeon.ink>2023-03-31 13:57:07 +0200
commit05b69cf768c60f2e9640d48fce98122656e27299 (patch)
treee4ae7baccc8431ecf1a4561061d5e8a1c8d64bb5 /train_ti.py
parentSupport Dadaptation d0, adjust sample freq when steps instead of epochs are used (diff)
downloadtextual-inversion-diff-05b69cf768c60f2e9640d48fce98122656e27299.tar.gz
textual-inversion-diff-05b69cf768c60f2e9640d48fce98122656e27299.tar.bz2
textual-inversion-diff-05b69cf768c60f2e9640d48fce98122656e27299.zip
Fix
Diffstat (limited to 'train_ti.py')
-rw-r--r--train_ti.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py
index 902f508..7900fbd 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -738,7 +738,7 @@ def main():
738 738
739 stats = list(zip(placeholder_tokens, placeholder_token_ids, initializer_tokens, initializer_token_ids)) 739 stats = list(zip(placeholder_tokens, placeholder_token_ids, initializer_tokens, initializer_token_ids))
740 740
741 print(f"{i + 1}: {stats})") 741 print(f"{i + 1}: {stats}")
742 742
743 datamodule = VlpnDataModule( 743 datamodule = VlpnDataModule(
744 data_file=args.train_data_file, 744 data_file=args.train_data_file,