1 files changed, 7 insertions, 0 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index a85ae4c..1b8a3d2 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -176,6 +176,12 @@ def parse_args():
         ),
     )
     parser.add_argument(
+        "--offset_noise_strength",
+        type=float,
+        default=0.15,
+        help="Perlin offset noise strength.",
+    )
+    parser.add_argument(
         "--num_train_epochs",
         type=int,
         default=100
@@ -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,