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_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') 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(): parser.add_argument( "--num_buckets", type=int, - default=0, + default=4, help="Number of aspect ratio buckets in either direction.", ) parser.add_argument( -- cgit v1.2.3-54-g00ecf