From 347ad308f8223d966793f0421c72432f7e912377 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 8 Feb 2023 11:38:56 +0100 Subject: Integrate Self-Attention-Guided (SAG) Stable Diffusion in my custom pipeline --- train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_dreambooth.py') 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(): parser.add_argument( "--adam_weight_decay", type=float, - default=1e-2, + default=0, help="Weight decay to use." ) parser.add_argument( -- cgit v1.2.3-54-g00ecf