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_ti.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_ti.py')
| -rw-r--r-- | train_ti.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/train_ti.py b/train_ti.py index ef71f6f..bbc5524 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -188,9 +188,9 @@ def parse_args(): | |||
| 188 | help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]', | 188 | help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]', |
| 189 | ) | 189 | ) |
| 190 | parser.add_argument( | 190 | parser.add_argument( |
| 191 | "--perlin_strength", | 191 | "--offset_noise_strength", |
| 192 | type=float, | 192 | type=float, |
| 193 | default=0.1, | 193 | default=0.15, |
| 194 | help="Perlin offset noise strength.", | 194 | help="Perlin offset noise strength.", |
| 195 | ) | 195 | ) |
| 196 | parser.add_argument( | 196 | parser.add_argument( |
| @@ -661,7 +661,7 @@ def main(): | |||
| 661 | checkpoint_frequency=args.checkpoint_frequency, | 661 | checkpoint_frequency=args.checkpoint_frequency, |
| 662 | milestone_checkpoints=not args.no_milestone_checkpoints, | 662 | milestone_checkpoints=not args.no_milestone_checkpoints, |
| 663 | global_step_offset=global_step_offset, | 663 | global_step_offset=global_step_offset, |
| 664 | perlin_strength=args.perlin_strength, | 664 | offset_noise_strength=args.offset_noise_strength, |
| 665 | # -- | 665 | # -- |
| 666 | tokenizer=tokenizer, | 666 | tokenizer=tokenizer, |
| 667 | sample_scheduler=sample_scheduler, | 667 | sample_scheduler=sample_scheduler, |