From 01f0b3bd5a7965776b420c97056f82601e2b7312 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 22 Jun 2023 18:34:53 +0200
Subject: Added prompt dropout

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

(limited to 'train_ti.py')

diff --git a/train_ti.py b/train_ti.py
index 7f93960..1dbd637 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -923,7 +923,7 @@ def main():
             progressive_buckets=args.progressive_buckets,
             bucket_step_size=args.bucket_step_size,
             bucket_max_pixels=args.bucket_max_pixels,
-            dropout=args.tag_dropout,
+            tag_dropout=args.tag_dropout,
             shuffle=not args.no_tag_shuffle,
             template_key=data_template,
             placeholder_tokens=args.placeholder_tokens,
-- 
cgit v1.2.3-54-g00ecf