diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 7 |
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(): | |||
176 | ), | 176 | ), |
177 | ) | 177 | ) |
178 | parser.add_argument( | 178 | parser.add_argument( |
179 | "--offset_noise_strength", | ||
180 | type=float, | ||
181 | default=0.15, | ||
182 | help="Perlin offset noise strength.", | ||
183 | ) | ||
184 | parser.add_argument( | ||
179 | "--num_train_epochs", | 185 | "--num_train_epochs", |
180 | type=int, | 186 | type=int, |
181 | default=100 | 187 | default=100 |
@@ -621,6 +627,7 @@ def main(): | |||
621 | num_train_epochs=args.num_train_epochs, | 627 | num_train_epochs=args.num_train_epochs, |
622 | gradient_accumulation_steps=args.gradient_accumulation_steps, | 628 | gradient_accumulation_steps=args.gradient_accumulation_steps, |
623 | sample_frequency=args.sample_frequency, | 629 | sample_frequency=args.sample_frequency, |
630 | offset_noise_strength=args.offset_noise_strength, | ||
624 | # -- | 631 | # -- |
625 | tokenizer=tokenizer, | 632 | tokenizer=tokenizer, |
626 | sample_scheduler=sample_scheduler, | 633 | sample_scheduler=sample_scheduler, |