diff options
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/train_lora.py b/train_lora.py index fa24cee..b16a99b 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -165,6 +165,12 @@ def parse_args(): | |||
165 | ), | 165 | ), |
166 | ) | 166 | ) |
167 | parser.add_argument( | 167 | parser.add_argument( |
168 | "--offset_noise_strength", | ||
169 | type=float, | ||
170 | default=0.15, | ||
171 | help="Perlin offset noise strength.", | ||
172 | ) | ||
173 | parser.add_argument( | ||
168 | "--num_train_epochs", | 174 | "--num_train_epochs", |
169 | type=int, | 175 | type=int, |
170 | default=100 | 176 | default=100 |
@@ -649,6 +655,7 @@ def main(): | |||
649 | num_train_epochs=args.num_train_epochs, | 655 | num_train_epochs=args.num_train_epochs, |
650 | gradient_accumulation_steps=args.gradient_accumulation_steps, | 656 | gradient_accumulation_steps=args.gradient_accumulation_steps, |
651 | sample_frequency=args.sample_frequency, | 657 | sample_frequency=args.sample_frequency, |
658 | offset_noise_strength=args.offset_noise_strength, | ||
652 | # -- | 659 | # -- |
653 | tokenizer=tokenizer, | 660 | tokenizer=tokenizer, |
654 | sample_scheduler=sample_scheduler, | 661 | sample_scheduler=sample_scheduler, |