summaryrefslogtreecommitdiffstats
path: root/train_lora.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-03-01 12:34:42 +0100
committerVolpeon <git@volpeon.ink>2023-03-01 12:34:42 +0100
commita1b8327085ddeab589be074d7e9df4291aba1210 (patch)
tree2f2016916d7a2f659268c3e375d55c59583c2b3b /train_lora.py
parentFixed TI normalization order (diff)
downloadtextual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.tar.gz
textual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.tar.bz2
textual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.zip
Update
Diffstat (limited to 'train_lora.py')
-rw-r--r--train_lora.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py
index d7c2de0..0a3d4c9 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -84,7 +84,7 @@ def parse_args():
84 parser.add_argument( 84 parser.add_argument(
85 "--num_buckets", 85 "--num_buckets",
86 type=int, 86 type=int,
87 default=4, 87 default=0,
88 help="Number of aspect ratio buckets in either direction.", 88 help="Number of aspect ratio buckets in either direction.",
89 ) 89 )
90 parser.add_argument( 90 parser.add_argument(