diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-08 11:38:56 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-08 11:38:56 +0100 |
| commit | 347ad308f8223d966793f0421c72432f7e912377 (patch) | |
| tree | 2b7319dc37787ce2828101c451987d086dd47360 /train_ti.py | |
| parent | Fixed Lora training (diff) | |
| download | textual-inversion-diff-347ad308f8223d966793f0421c72432f7e912377.tar.gz textual-inversion-diff-347ad308f8223d966793f0421c72432f7e912377.tar.bz2 textual-inversion-diff-347ad308f8223d966793f0421c72432f7e912377.zip | |
Integrate Self-Attention-Guided (SAG) Stable Diffusion in my custom pipeline
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 2840def..c79dfa2 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -143,7 +143,7 @@ def parse_args(): | |||
| 143 | parser.add_argument( | 143 | parser.add_argument( |
| 144 | "--num_buckets", | 144 | "--num_buckets", |
| 145 | type=int, | 145 | type=int, |
| 146 | default=0, | 146 | default=4, |
| 147 | help="Number of aspect ratio buckets in either direction.", | 147 | help="Number of aspect ratio buckets in either direction.", |
| 148 | ) | 148 | ) |
| 149 | parser.add_argument( | 149 | parser.add_argument( |