From d9bb4a0d43276c8e120866af044fcf3566930859 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 23 Mar 2023 22:15:17 +0100
Subject: Bring back Perlin offset noise

---
 train_ti.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'train_ti.py')

diff --git a/train_ti.py b/train_ti.py
index 9bc74c1..ef71f6f 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -187,6 +187,12 @@ def parse_args():
         default="auto",
         help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]',
     )
+    parser.add_argument(
+        "--perlin_strength",
+        type=float,
+        default=0.1,
+        help="Perlin offset noise strength.",
+    )
     parser.add_argument(
         "--num_train_epochs",
         type=int,
@@ -655,6 +661,7 @@ def main():
         checkpoint_frequency=args.checkpoint_frequency,
         milestone_checkpoints=not args.no_milestone_checkpoints,
         global_step_offset=global_step_offset,
+        perlin_strength=args.perlin_strength,
         # --
         tokenizer=tokenizer,
         sample_scheduler=sample_scheduler,
-- 
cgit v1.2.3-54-g00ecf