summaryrefslogtreecommitdiffstats
path: root/train_ti.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_ti.py')
-rw-r--r--train_ti.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py
index 68783ea..394711f 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -607,7 +607,7 @@ def main():
607 with_prior_preservation=args.num_class_images != 0, 607 with_prior_preservation=args.num_class_images != 0,
608 prior_loss_weight=args.prior_loss_weight, 608 prior_loss_weight=args.prior_loss_weight,
609 no_val=args.valid_set_size == 0, 609 no_val=args.valid_set_size == 0,
610 low_freq_noise=0, 610 # low_freq_noise=0,
611 strategy=textual_inversion_strategy, 611 strategy=textual_inversion_strategy,
612 num_train_epochs=args.num_train_epochs, 612 num_train_epochs=args.num_train_epochs,
613 gradient_accumulation_steps=args.gradient_accumulation_steps, 613 gradient_accumulation_steps=args.gradient_accumulation_steps,