From 27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sat, 24 Jun 2023 16:26:22 +0200
Subject: Fixes

---
 train_lora.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'train_lora.py')

diff --git a/train_lora.py b/train_lora.py
index fbec009..2a43252 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -235,6 +235,12 @@ def parse_args():
         nargs="*",
         help="A collection to filter the dataset.",
     )
+    parser.add_argument(
+        "--validation_prompts",
+        type=str,
+        nargs="*",
+        help="Prompts for additional validation images",
+    )
     parser.add_argument(
         "--seed", type=int, default=None, help="A seed for reproducible training."
     )
@@ -545,7 +551,7 @@ def parse_args():
     parser.add_argument(
         "--sample_steps",
         type=int,
-        default=10,
+        default=15,
         help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.",
     )
     parser.add_argument(
-- 
cgit v1.2.3-70-g09d2