summaryrefslogtreecommitdiffstats
path: root/train_ti.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-08 20:36:17 +0100
committerVolpeon <git@volpeon.ink>2023-01-08 20:36:17 +0100
commited892a06ba7a231a84d47bd835fc625aa3f2c75c (patch)
tree4dc33b2b0bebcf6b7880a42e0518e4ae5eecd3ad /train_ti.py
parentImproved aspect ratio bucketing (diff)
downloadtextual-inversion-diff-ed892a06ba7a231a84d47bd835fc625aa3f2c75c.tar.gz
textual-inversion-diff-ed892a06ba7a231a84d47bd835fc625aa3f2c75c.tar.bz2
textual-inversion-diff-ed892a06ba7a231a84d47bd835fc625aa3f2c75c.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 eb0b8b6..03f52c4 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -142,7 +142,7 @@ def parse_args():
142 parser.add_argument( 142 parser.add_argument(
143 "--num_buckets", 143 "--num_buckets",
144 type=int, 144 type=int,
145 default=4, 145 default=0,
146 help="Number of aspect ratio buckets in either direction.", 146 help="Number of aspect ratio buckets in either direction.",
147 ) 147 )
148 parser.add_argument( 148 parser.add_argument(