diff options
Diffstat (limited to 'train_lora.py')
| -rw-r--r-- | train_lora.py | 2 |
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( |