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( |