diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-23 21:53:46 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-23 21:53:46 +0100 |
| commit | b491a817088790219e052b86173e128c55b597f8 (patch) | |
| tree | 49ea8480da78b002f36b15b52d1b9bc501837819 /train_dreambooth.py | |
| parent | Improved class prompt handling (diff) | |
| download | textual-inversion-diff-b491a817088790219e052b86173e128c55b597f8.tar.gz textual-inversion-diff-b491a817088790219e052b86173e128c55b597f8.tar.bz2 textual-inversion-diff-b491a817088790219e052b86173e128c55b597f8.zip | |
num_class_images is now class images per train image
Diffstat (limited to 'train_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
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(): | |||
| 101 | parser.add_argument( | 101 | parser.add_argument( |
| 102 | "--num_class_images", | 102 | "--num_class_images", |
| 103 | type=int, | 103 | type=int, |
| 104 | default=400, | 104 | default=1, |
| 105 | help="How many class images to generate." | 105 | help="How many class images to generate." |
| 106 | ) | 106 | ) |
| 107 | parser.add_argument( | 107 | parser.add_argument( |