diff options
author | Volpeon <git@volpeon.ink> | 2023-06-22 18:34:53 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-06-22 18:34:53 +0200 |
commit | 01f0b3bd5a7965776b420c97056f82601e2b7312 (patch) | |
tree | 236bde99b6116abdc7ca75fb57828ce21b74ba32 /train_lora.py | |
parent | Update (diff) | |
download | textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.tar.gz textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.tar.bz2 textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.zip |
Added prompt dropout
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index 51dc827..1ff25ff 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -1137,7 +1137,7 @@ def main(): | |||
1137 | lora_datamodule = create_datamodule( | 1137 | lora_datamodule = create_datamodule( |
1138 | valid_set_size=args.valid_set_size, | 1138 | valid_set_size=args.valid_set_size, |
1139 | batch_size=args.train_batch_size, | 1139 | batch_size=args.train_batch_size, |
1140 | dropout=args.tag_dropout, | 1140 | tag_dropout=args.tag_dropout, |
1141 | filter=partial(keyword_filter, None, args.collection, args.exclude_collections), | 1141 | filter=partial(keyword_filter, None, args.collection, args.exclude_collections), |
1142 | ) | 1142 | ) |
1143 | lora_datamodule.setup() | 1143 | lora_datamodule.setup() |