From 19ae465203c8dcc0b1179584db632015362b5e44 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sun, 26 Mar 2023 14:27:54 +0200
Subject: Improved inverted tokens

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

(limited to 'train_dreambooth.py')

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 7a33bca..9345797 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -585,6 +585,7 @@ def main():
         dropout=args.tag_dropout,
         shuffle=not args.no_tag_shuffle,
         template_key=args.train_data_template,
+        placeholder_tokens=args.placeholder_tokens,
         valid_set_size=args.valid_set_size,
         train_set_pad=args.train_set_pad,
         valid_set_pad=args.valid_set_pad,
-- 
cgit v1.2.3-54-g00ecf