diff options
author | Volpeon <git@volpeon.ink> | 2023-03-24 17:23:09 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-03-24 17:23:09 +0100 |
commit | 9bfb4a078f63a7ce6e35e89093f17febd9ff4b51 (patch) | |
tree | 41b83780c79803531c7208a72bff9206ffa908da /train_dreambooth.py | |
parent | Fixed Lora training perf issue (diff) | |
download | textual-inversion-diff-9bfb4a078f63a7ce6e35e89093f17febd9ff4b51.tar.gz textual-inversion-diff-9bfb4a078f63a7ce6e35e89093f17febd9ff4b51.tar.bz2 textual-inversion-diff-9bfb4a078f63a7ce6e35e89093f17febd9ff4b51.zip |
Update
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, |