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_lora.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train_lora.py')

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():
     lora_datamodule = create_datamodule(
         valid_set_size=args.valid_set_size,
         batch_size=args.train_batch_size,
-        dropout=args.tag_dropout,
+        tag_dropout=args.tag_dropout,
         filter=partial(keyword_filter, None, args.collection, args.exclude_collections),
     )
     lora_datamodule.setup()
-- 
cgit v1.2.3-54-g00ecf