From b491a817088790219e052b86173e128c55b597f8 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 23 Dec 2022 21:53:46 +0100
Subject: num_class_images is now class images per train image

---
 train_dreambooth.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train_dreambooth.py')

diff --git a/train_dreambooth.py b/train_dreambooth.py
index ff67d12..2f913e7 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -101,7 +101,7 @@ def parse_args():
     parser.add_argument(
         "--num_class_images",
         type=int,
-        default=400,
+        default=1,
         help="How many class images to generate."
     )
     parser.add_argument(
-- 
cgit v1.2.3-54-g00ecf