From 9bfb4a078f63a7ce6e35e89093f17febd9ff4b51 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 24 Mar 2023 17:23:09 +0100 Subject: Update --- train_dreambooth.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index a85ae4c..1b8a3d2 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -175,6 +175,12 @@ def parse_args(): " resolution" ), ) + parser.add_argument( + "--offset_noise_strength", + type=float, + default=0.15, + help="Perlin offset noise strength.", + ) parser.add_argument( "--num_train_epochs", type=int, @@ -621,6 +627,7 @@ def main(): num_train_epochs=args.num_train_epochs, gradient_accumulation_steps=args.gradient_accumulation_steps, sample_frequency=args.sample_frequency, + offset_noise_strength=args.offset_noise_strength, # -- tokenizer=tokenizer, sample_scheduler=sample_scheduler, -- cgit v1.2.3-54-g00ecf