diff options
author | Volpeon <git@volpeon.ink> | 2022-10-13 21:11:53 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-10-13 21:11:53 +0200 |
commit | 515f0f1fdc9a76bf63bd746c291dcfec7fc747fb (patch) | |
tree | cf4bbf1cae822cf7c7f388b6918154032def0376 /data | |
parent | Added TI+Dreambooth training (diff) | |
download | textual-inversion-diff-515f0f1fdc9a76bf63bd746c291dcfec7fc747fb.tar.gz textual-inversion-diff-515f0f1fdc9a76bf63bd746c291dcfec7fc747fb.tar.bz2 textual-inversion-diff-515f0f1fdc9a76bf63bd746c291dcfec7fc747fb.zip |
Added support for Aesthetic Gradients
Diffstat (limited to 'data')
-rw-r--r-- | data/csv.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/csv.py b/data/csv.py index 253ce9e..aad970c 100644 --- a/data/csv.py +++ b/data/csv.py | |||
@@ -23,7 +23,7 @@ class CSVDataModule(pl.LightningDataModule): | |||
23 | tokenizer, | 23 | tokenizer, |
24 | instance_identifier, | 24 | instance_identifier, |
25 | class_identifier=None, | 25 | class_identifier=None, |
26 | class_subdir="db_cls", | 26 | class_subdir="cls", |
27 | num_class_images=100, | 27 | num_class_images=100, |
28 | size=512, | 28 | size=512, |
29 | repeats=100, | 29 | repeats=100, |