diff options
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index fbec009..2a43252 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -236,6 +236,12 @@ def parse_args(): | |||
236 | help="A collection to filter the dataset.", | 236 | help="A collection to filter the dataset.", |
237 | ) | 237 | ) |
238 | parser.add_argument( | 238 | parser.add_argument( |
239 | "--validation_prompts", | ||
240 | type=str, | ||
241 | nargs="*", | ||
242 | help="Prompts for additional validation images", | ||
243 | ) | ||
244 | parser.add_argument( | ||
239 | "--seed", type=int, default=None, help="A seed for reproducible training." | 245 | "--seed", type=int, default=None, help="A seed for reproducible training." |
240 | ) | 246 | ) |
241 | parser.add_argument( | 247 | parser.add_argument( |
@@ -545,7 +551,7 @@ def parse_args(): | |||
545 | parser.add_argument( | 551 | parser.add_argument( |
546 | "--sample_steps", | 552 | "--sample_steps", |
547 | type=int, | 553 | type=int, |
548 | default=10, | 554 | default=15, |
549 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", | 555 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", |
550 | ) | 556 | ) |
551 | parser.add_argument( | 557 | parser.add_argument( |