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_dreambooth.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_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index a29c507..8ac70e8 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -304,7 +304,7 @@ def parse_args(): | |||
| 304 | parser.add_argument( | 304 | parser.add_argument( |
| 305 | "--adam_weight_decay", | 305 | "--adam_weight_decay", |
| 306 | type=float, | 306 | type=float, |
| 307 | default=1e-2, | 307 | default=0, |
| 308 | help="Weight decay to use." | 308 | help="Weight decay to use." |
| 309 | ) | 309 | ) |
| 310 | parser.add_argument( | 310 | parser.add_argument( |