1 files changed, 1 insertions, 0 deletions

diff --git a/train_ti.py b/train_ti.py
index 171d085..f78c7d2 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -586,6 +586,7 @@ def main():
         seed=args.seed,
         with_prior_preservation=args.num_class_images != 0,
         prior_loss_weight=args.prior_loss_weight,
+        low_freq_noise=0,
         strategy=textual_inversion_strategy,
         num_train_epochs=args.num_train_epochs,
         sample_frequency=args.sample_frequency,